diff --git a/utils/logger.py b/utils/logger.py
index d643f36a..45370d3d 100644
--- a/utils/logger.py
+++ b/utils/logger.py
@@ -1,7 +1,13 @@
 import logging
+from os import PathLike
+from typing import Optional, Union
 
 
-def setup_logger(log_file=None, level=logging.INFO, overwrite_handler=False):
+def setup_logger(
+    log_file: Optional[Union[str, PathLike[str]]] = None,
+    level: int = logging.INFO,
+    overwrite_handler: bool = False,
+) -> logging.Logger:
     # Create a logger and set the logging level
     logger = logging.getLogger()
     logger.setLevel(level)