nshtrainer 0.10.9__tar.gz → 0.10.10__tar.gz

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.10.9
+Version: 0.10.10
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "0.10.9"
+version = "0.10.10"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/early_stopping.py

@@ -1,5 +1,5 @@
+import logging
 import math
-from logging import getLogger
 
 from lightning.fabric.utilities.rank_zero import _get_rank
 from lightning.pytorch import Trainer
@@ -7,7 +7,7 @@ from lightning.pytorch.callbacks import EarlyStopping as _EarlyStopping
 from lightning.pytorch.utilities.rank_zero import rank_prefixed_message
 from typing_extensions import override
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 class EarlyStopping(_EarlyStopping):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/finite_checks.py

@@ -1,4 +1,4 @@
-from logging import getLogger
+import logging
 from typing import Literal
 
 import torch
@@ -7,7 +7,7 @@ from typing_extensions import override
 
 from .base import CallbackConfigBase
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def finite_checks(

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/model_checkpoint.py

@@ -1,6 +1,6 @@
+import logging
 import re
 from datetime import timedelta
-from logging import getLogger
 from pathlib import Path
 from typing import TYPE_CHECKING, Literal
 
@@ -15,7 +15,7 @@ from .base import CallbackConfigBase
 if TYPE_CHECKING:
     from ..model.config import BaseConfig
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def _convert_string(input_string: str):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/norm_logging.py

@@ -1,4 +1,4 @@
-from logging import getLogger
+import logging
 from typing import Literal, cast
 
 import torch
@@ -9,7 +9,7 @@ from typing_extensions import override
 
 from .base import CallbackConfigBase
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def grad_norm(

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/throughput_monitor.py

@@ -1,4 +1,4 @@
-from logging import getLogger
+import logging
 from typing import Any, Literal, Protocol, TypedDict, cast, runtime_checkable
 
 from typing_extensions import NotRequired, override
@@ -6,7 +6,7 @@ from typing_extensions import NotRequired, override
 from ._throughput_monitor_callback import ThroughputMonitor as _ThroughputMonitor
 from .base import CallbackConfigBase
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 class ThroughputMonitorBatchStats(TypedDict):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/callbacks/wandb_watch.py

@@ -1,4 +1,4 @@
-from logging import getLogger
+import logging
 from typing import Literal, Protocol, cast, runtime_checkable
 
 import torch.nn as nn
@@ -9,7 +9,7 @@ from typing_extensions import override
 
 from .base import CallbackConfigBase
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 @runtime_checkable

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/data/balanced_batch_sampler.py

@@ -1,6 +1,6 @@
 import heapq
+import logging
 from functools import cached_property
-from logging import getLogger
 from typing import Any, Protocol, runtime_checkable
 
 import numpy as np
@@ -10,7 +10,7 @@ from lightning_fabric.utilities.distributed import _DatasetSamplerWrapper
 from torch.utils.data import BatchSampler, Dataset, DistributedSampler
 from typing_extensions import override
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def _all_gather(tensor: torch.Tensor, device: torch.device | None = None):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/base.py

@@ -1,7 +1,7 @@
 import inspect
+import logging
 from abc import ABC, abstractmethod
 from collections.abc import MutableMapping
-from logging import getLogger
 from typing import IO, TYPE_CHECKING, Any, Generic, cast
 
 import torch
@@ -21,7 +21,7 @@ from .modules.profiler import ProfilerMixin
 from .modules.rlp_sanity_checks import RLPSanityCheckModuleMixin
 from .modules.shared_parameters import SharedParametersModuleMixin
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 THparams = TypeVar("THparams", bound=BaseConfig, infer_variance=True)
 

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/config.py

@@ -1,4 +1,5 @@
 import copy
+import logging
 import os
 import string
 import time
@@ -6,7 +7,6 @@ import warnings
 from abc import ABC, abstractmethod
 from collections.abc import Iterable, Sequence
 from datetime import timedelta
-from logging import getLogger
 from pathlib import Path
 from typing import (
     Annotated,
@@ -46,7 +46,7 @@ from ..callbacks.base import CallbackConfigBase
 from ..metrics import MetricConfig
 from ._environment import EnvironmentConfig
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 class IdSeedWarning(Warning):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/modules/callback.py

@@ -1,6 +1,6 @@
+import logging
 from collections import abc
 from collections.abc import Callable, Iterable
-from logging import getLogger
 from typing import Any, TypeAlias, cast, final
 
 from lightning.pytorch import Callback, LightningModule
@@ -9,7 +9,7 @@ from typing_extensions import override
 
 from ...util.typing_utils import mixin_base_type
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 CallbackFn: TypeAlias = Callable[[], Callback | Iterable[Callback] | None]
 

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/modules/debug.py

@@ -1,9 +1,9 @@
-from logging import getLogger
+import logging
 
 import torch
 import torch.distributed
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 class DebugModuleMixin:

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/modules/rlp_sanity_checks.py

@@ -1,5 +1,5 @@
+import logging
 from collections.abc import Mapping
-from logging import getLogger
 from typing import cast
 
 import torch
@@ -14,7 +14,7 @@ from ...util.typing_utils import mixin_base_type
 from ..config import BaseConfig
 from .callback import CallbackModuleMixin
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def _on_train_start_callback(trainer: Trainer, pl_module: LightningModule):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/model/modules/shared_parameters.py

@@ -1,5 +1,5 @@
+import logging
 from collections.abc import Sequence
-from logging import getLogger
 from typing import cast
 
 import torch.nn as nn
@@ -10,7 +10,7 @@ from ...util.typing_utils import mixin_base_type
 from ..config import BaseConfig
 from .callback import CallbackRegistrarModuleMixin
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def _parameters_to_names(parameters: Sequence[nn.Parameter], model: nn.Module):

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/util/environment.py

@@ -1,8 +1,8 @@
+import logging
 import os
 from contextlib import contextmanager
-from logging import getLogger
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 @contextmanager

{nshtrainer-0.10.9 → nshtrainer-0.10.10}/src/nshtrainer/util/seed.py

@@ -1,8 +1,8 @@
-from logging import getLogger
+import logging
 
 import lightning.fabric.utilities.seed as LS
 
-log = getLogger(__name__)
+log = logging.getLogger(__name__)
 
 
 def seed_everything(seed: int | None, *, workers: bool = False):