linting

molpipeline/mol2any/mol2concatinated_vector.py

@@ -11,9 +11,8 @@
 
 import numpy as np
 import numpy.typing as npt
-from sklearn.base import clone
-
 from loguru import logger
+from sklearn.base import clone
 
 from molpipeline.abstract_pipeline_elements.core import (
     InvalidInstance,

tests/utils/logging.py

@@ -1,11 +1,15 @@
 """Test utils for logging."""
+
 from contextlib import contextmanager
 from typing import Generator
 
 from loguru import logger
 
+
 @contextmanager
-def capture_logs(level="INFO", format="{level}:{name}:{message}") -> Generator[list[str], None, None]:
+def capture_logs(
+    level="INFO", format="{level}:{name}:{message}"
+) -> Generator[list[str], None, None]:
     """Capture loguru-based logs.
 
     Custom context manager to test loguru-based logs. For details and usage examples,
@@ -17,8 +21,13 @@ def capture_logs(level="INFO", format="{level}:{name}:{message}") -> Generator[l
         Log level, by default "INFO"
     format : str, optional
         Log format, by default "{level}:{name}:{message}"
+
+    Yields
+    -------
+    list[str]
+        List of log messages
     """
     output: list[str] = []
     handler_id = logger.add(output.append, level=level, format=format)
     yield output
-    logger.remove(handler_id)
+    logger.remove(handler_id)