Skip to content

Commit 47c61d9

Browse files
fix: add type hints to logger method arguments for better clarity
1 parent a4b7777 commit 47c61d9

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

mct_quantizers/logger.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __check_path_create_dir(log_path: str) -> None:
3838
Create a path if not exist. Otherwise, do nothing.
3939
4040
Args:
41-
log_path: Path to create or verify that exists.
41+
log_path (str): Path to create or verify that exists.
4242
"""
4343

4444
if not os.path.exists(log_path):
@@ -50,7 +50,7 @@ def set_logger_level(log_level: int = logging.INFO) -> None:
5050
Set log level to determine the logger verbosity.
5151
5252
Args:
53-
log_level: Level of verbosity to set for the logger.
53+
log_level (int): Level of verbosity to set for the logger.
5454
"""
5555

5656
logger = Logger.get_logger()
@@ -62,7 +62,7 @@ def set_handler_level(log_level: int = logging.INFO) -> None:
6262
Set log level for all handlers attached to the logger.
6363
6464
Args:
65-
log_level: Level of verbosity to set for the handlers.
65+
log_level (int): Level of verbosity to set for the handlers.
6666
"""
6767

6868
logger = Logger.get_logger()
@@ -99,7 +99,7 @@ def set_log_file(log_folder: Optional[str] = None) -> None:
9999
In that folder, it creates a log file according to the timestamp.
100100
101101
Args:
102-
log_folder: Folder path to hold the log file.
102+
log_folder (Optional[str]): Folder path to hold the log file.
103103
"""
104104

105105
logger = Logger.get_logger()
@@ -137,7 +137,7 @@ def critical(msg: str) -> None:
137137
Log a message at 'critical' severity and raise an exception.
138138
139139
Args:
140-
msg: Message to log.
140+
msg (str): Message to log.
141141
"""
142142
Logger.get_logger().critical(msg)
143143
raise Exception(msg)
@@ -148,7 +148,7 @@ def exception(msg: str) -> None:
148148
Log a message at 'exception' severity and raise an exception.
149149
150150
Args:
151-
msg: Message to log.
151+
msg (str): Message to log.
152152
"""
153153
Logger.get_logger().exception(msg)
154154
raise Exception(msg)
@@ -159,7 +159,7 @@ def debug(msg: str) -> None:
159159
Log a message at 'debug' severity.
160160
161161
Args:
162-
msg: Message to log.
162+
msg (str): Message to log.
163163
"""
164164
Logger.get_logger().debug(msg)
165165

@@ -169,7 +169,7 @@ def info(msg: str) -> None:
169169
Log a message at 'info' severity.
170170
171171
Args:
172-
msg: Message to log.
172+
msg (str): Message to log.
173173
"""
174174
Logger.get_logger().info(msg)
175175

@@ -179,7 +179,7 @@ def warning(msg: str) -> None:
179179
Log a message at 'warning' severity.
180180
181181
Args:
182-
msg: Message to log.
182+
msg (str): Message to log.
183183
"""
184184
Logger.get_logger().warning(msg)
185185

@@ -189,7 +189,7 @@ def error(msg: str) -> None:
189189
Log a message at 'error' severity and raise an exception.
190190
191191
Args:
192-
msg: Message to log.
192+
msg (str): Message to log.
193193
"""
194194
Logger.get_logger().error(msg)
195195

@@ -199,8 +199,8 @@ def set_log_folder(folder: str, level: int = logging.INFO) -> None:
199199
Set a directory path for saving a log file.
200200
201201
Args:
202-
folder: Folder path to save the log file.
203-
level: Level of verbosity to set to the logger and handlers.
202+
folder (str): Folder path to save the log file.
203+
level (int): Level of verbosity to set to the logger and handlers.
204204
205205
Note:
206206
This is a convenience function that calls multiple Logger methods

0 commit comments

Comments
 (0)