diff --git a/src/neptune/cli/sync.py b/src/neptune/cli/sync.py index a96d54064..bed717ad6 100644 --- 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 index efa5a5c88..462b568db 100644 --- 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 index 9217ba401..ffc6ee93a 100644 --- 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 index 922513477..c19baa19a 100644 --- 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