logger done! need to add node2vec tests

Author: Ashutosh Tiwari

graph_ml/models/__init__.py

@@ -1,2 +1,2 @@
-from ..models.torch_node2vec import TorchNode2Vec
 from ..models.gensim_node2vec import GensimNode2Vec
+from ..models.torch_node2vec import TorchNode2Vec

graph_ml/utils/logger.py

@@ -1,30 +1,15 @@
-import logging
 import asyncio
-from logging import getLogger, StreamHandler, FileHandler, Formatter
-from concurrent.futures import ThreadPoolExecutor
+import logging
+from logging import Formatter, StreamHandler, getLogger
+
 from colorama import Fore, Style, init
+
 from ..utils import utils
 
 # Initialize colorama for Windows compatibility
 init(autoreset=True)
 
 
-class AsyncFileHandler(FileHandler):
-    def __init__(self, filename):
-        super().__init__(filename)
-        self.loop = asyncio.get_event_loop()
-        self.executor = ThreadPoolExecutor()
-
-    async def emit(self, record):
-        msg = self.format(record)
-        await self.loop.run_in_executor(self.executor, self._write, msg)
-
-    def _write(self, msg):
-        with self._open() as file:
-            file.write(f"{msg}\n")
-            file.flush()
-
-
 class ColorFormatter(Formatter):
     """Custom formatter to add colors depending on the log level."""
 
@@ -49,7 +34,7 @@ def __new__(cls, *args, **kwargs):
             cls._instance = super(AsyncLogger, cls).__new__(cls)
         return cls._instance
 
-    def __init__(self, name, file_path=None):
+    def __init__(self, name):
         if hasattr(self, "_initialized") and self._initialized:
             return  # Prevent re-initialization
         self._initialized = True
@@ -67,14 +52,6 @@ def __init__(self, name, file_path=None):
         console_handler.setFormatter(formatter)
         self.logger.addHandler(console_handler)
 
-        # File Handler (Asynchronous)
-        if file_path:
-            async_file_handler = AsyncFileHandler(file_path)
-            async_file_handler.setFormatter(
-                Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-            )  # No color for file
-            self.logger.addHandler(async_file_handler)
-
     async def info(self, message):
         await self._log_async(self.logger.info, message)
 
@@ -116,5 +93,5 @@ def debug(self, message):
 
 
 # Exposing the logger as a singleton instance
-logger_instance = AsyncLogger(name="graph_ml", file_path="graph_ml.log")
+logger_instance = AsyncLogger(name="graph_ml")
 logger = LoggerWrapper(logger_instance)

graph_ml/utils/utils.py

@@ -1,10 +1,9 @@
+import asyncio
 import os
 
 import numpy as np
 from numba import njit
 from scipy import sparse
-import asyncio
-
 
 # no global variables in this module
 

tests/test_logger.py

@@ -1,12 +1,8 @@
 from graph_ml.utils.logger import logger
-from os import path
 
 
-def test_output_file():
+def test_logger():
     logger.info("Test info message")
     logger.error("Test error message")
     logger.debug("Test debug message")
-    # check output file graph_ml.log exists
-    assert path.exists("graph_ml.log")
-    # check output file is not empty
-    assert path.getsize("graph_ml.log") > 0
+    assert True