nshtrainer 0.10.11__tar.gz → 0.10.12__tar.gz

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/PKG-INFO

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

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/pyproject.toml

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

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/src/nshtrainer/_checkpoint/metadata.py

@@ -10,7 +10,7 @@ import nshconfig as C
 import numpy as np
 import torch
 
-from ..model._environment import EnvironmentConfig
+from ..util._environment_info import EnvironmentConfig
 
 if TYPE_CHECKING:
     from ..model import BaseConfig, LightningModuleBase

nshtrainer-0.10.12/src/nshtrainer/ll/model.py

@@ -0,0 +1,12 @@
+from nshtrainer.model import *  # noqa: F403
+
+from ..util._environment_info import (
+    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
+)
+from ..util._environment_info import EnvironmentConfig as EnvironmentConfig
+from ..util._environment_info import (
+    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
+)
+from ..util._environment_info import (
+    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
+)

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/src/nshtrainer/model/__init__.py

@@ -1,16 +1,5 @@
 from typing_extensions import TypeAlias
 
-from ._environment import (
-    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
-)
-from ._environment import EnvironmentConfig as EnvironmentConfig
-from ._environment import (
-    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
-)
-from ._environment import (
-    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
-)
-from ._environment import EnvironmentSnapshotConfig as EnvironmentSnapshotConfig
 from .base import Base as Base
 from .base import LightningModuleBase as LightningModuleBase
 from .config import BaseConfig as BaseConfig

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/src/nshtrainer/model/base.py

@@ -11,7 +11,7 @@ from lightning.pytorch.callbacks import Callback
 from lightning.pytorch.utilities.types import STEP_OUTPUT
 from typing_extensions import Self, TypeVar, override
 
-from ._environment import EnvironmentConfig
+from ..util._environment_info import EnvironmentConfig
 from .config import BaseConfig
 from .modules.callback import CallbackModuleMixin
 from .modules.debug import DebugModuleMixin

{nshtrainer-0.10.11 → nshtrainer-0.10.12}/src/nshtrainer/model/config.py

@@ -44,7 +44,7 @@ from ..callbacks import (
 )
 from ..callbacks.base import CallbackConfigBase
 from ..metrics import MetricConfig
-from ._environment import EnvironmentConfig
+from ..util._environment_info import EnvironmentConfig
 
 log = logging.getLogger(__name__)
 

nshtrainer-0.10.11/src/nshtrainer/model/_environment.py → nshtrainer-0.10.12/src/nshtrainer/util/_environment_info.py

@@ -14,11 +14,11 @@ import psutil
 import torch
 from typing_extensions import Self
 
-from ..util.slurm import parse_slurm_node_list
+from .slurm import parse_slurm_node_list
 
 if TYPE_CHECKING:
-    from .base import LightningModuleBase
-    from .config import BaseConfig
+    from ..model.base import LightningModuleBase
+    from ..model.config import BaseConfig
 
 
 log = logging.getLogger(__name__)

nshtrainer-0.10.11/src/nshtrainer/ll/model.py

@@ -1 +0,0 @@
-from nshtrainer.model import *  # noqa: F403