small bugfix

neptune-ai · Jan 11, 2024 · ab28e41 · ab28e41
1 parent db2d036
commit ab28e41
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/src/neptune/cli/sync.py b/src/neptune/cli/sync.py
@@ -186,7 +186,7 @@ def _register_offline_container(self, project: Project, container_type: Containe
             logger.warning(
                 "Exception occurred while trying to create a run" " on the Neptune server. Please try again later",
             )
-            logging.exception(e)
+            logger.exception(e)
             return None
 
     @staticmethod

diff --git a/src/neptune/cli/utils.py b/src/neptune/cli/utils.py
@@ -75,7 +75,7 @@ def get_metadata_container(
         logger.warning("Can't fetch %s %s. Skipping.", public_container_type, container_id)
     except NeptuneException as e:
         logger.warning("Exception while fetching %s %s. Skipping.", public_container_type, container_id)
-        logging.exception(e)
+        logger.exception(e)
 
     return None
 

diff --git a/src/neptune/internal/backends/utils.py b/src/neptune/internal/backends/utils.py
@@ -79,7 +79,7 @@
 from neptune.internal.utils import replace_patch_version
 from neptune.internal.utils.logger import get_logger
 
-_logger = get_logger()
+logger = get_logger()
 
 if TYPE_CHECKING:
     from neptune.internal.backends.neptune_backend import NeptuneBackend

diff --git a/tests/unit/neptune/new/test_logging.py b/tests/unit/neptune/new/test_logging.py
@@ -1,5 +1,4 @@
 from contextlib import contextmanager
-from typing import Optional
 
 import pytest