Remove logging unless debug mode is set

dltype/_lib/_core.py

@@ -4,7 +4,6 @@
 
 import inspect
 import itertools
-import logging
 import warnings
 from functools import lru_cache, wraps
 from typing import (
@@ -31,14 +30,15 @@
     _dltype_context,
     _dtypes,
     _errors,
+    _log_utils,
     _tensor_type_base,
 )
 
 if TYPE_CHECKING:
     from collections.abc import Callable
 
 
-_logger: Final = logging.getLogger(__name__)
+_logger: Final = _log_utils.get_logger(__name__)
 
 P = ParamSpec("P")
 R = TypeVar("R")

dltype/_lib/_dltype_context.py

@@ -2,15 +2,14 @@
 
 from __future__ import annotations
 
-import logging
 import time
 import warnings
 from collections import deque
 from typing import Any, Final, NamedTuple, TypeAlias, cast
 
-from dltype._lib import _constants, _dtypes, _errors, _parser, _tensor_type_base
+from dltype._lib import _constants, _dtypes, _errors, _log_utils, _parser, _tensor_type_base
 
-_logger: Final = logging.getLogger(__name__)
+_logger: Final = _log_utils.get_logger(__name__)
 
 EvaluatedDimensionT: TypeAlias = dict[str, int]
 

dltype/_lib/_log_utils.py

@@ -0,0 +1,31 @@
+"""Utilities for logging in the dltype library."""
+
+import logging
+import typing
+
+from dltype._lib import _constants
+
+
+class DummyLogger:
+    def debug(self, *args: typing.Any) -> None:  # noqa: ANN401
+        pass
+
+    def info(self, *args: typing.Any) -> None:  # noqa: ANN401
+        pass
+
+    def warning(self, *args: typing.Any) -> None:  # noqa: ANN401
+        pass
+
+    def error(self, *args: typing.Any) -> None:  # noqa: ANN401
+        pass
+
+    def exception(self, *args: typing.Any) -> None:  # noqa: ANN401
+        pass
+
+
+def get_logger(name: str) -> logging.Logger | DummyLogger:
+    if _constants.GLOBAL_DISABLE:
+        return DummyLogger()
+    if not _constants.DEBUG_MODE:
+        return DummyLogger()
+    return logging.getLogger(name)

dltype/_lib/_parser.py

@@ -3,14 +3,15 @@
 from __future__ import annotations
 
 import enum
-import logging
 import math
 import re
 import typing
 
 from typing_extensions import override
 
-_logger: typing.Final = logging.getLogger(__name__)
+from dltype._lib import _log_utils
+
+_logger: typing.Final = _log_utils.get_logger(__name__)
 
 
 class _DLTypeSpecifier(enum.Enum):

dltype/_lib/_tensor_type_base.py

@@ -15,9 +15,7 @@
     _parser,
     _symbolic_expressions,
 )
-from dltype._lib import (
-    _dependency_utilities as _deps,
-)
+from dltype._lib import _dependency_utilities as _deps
 
 if typing.TYPE_CHECKING:
     from pydantic import GetCoreSchemaHandler, ValidationInfo

pyproject.toml

@@ -34,7 +34,7 @@ license-files = ["LICENSE"]
 name = "dltype"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.11.0"
+version = "0.12.0"
 
 [project.optional-dependencies]
 jax = ["jax>=0.6.2"]