nshtrainer 0.8.2__tar.gz → 0.8.4__tar.gz

{nshtrainer-0.8.2 → nshtrainer-0.8.4}/PKG-INFO

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

{nshtrainer-0.8.2 → nshtrainer-0.8.4}/pyproject.toml

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

nshtrainer-0.8.4/src/nshtrainer/__init__.py

@@ -0,0 +1,14 @@
+from . import _experimental as _experimental
+from . import callbacks as callbacks
+from . import data as data
+from . import lr_scheduler as lr_scheduler
+from . import model as model
+from . import nn as nn
+from . import optimizer as optimizer
+from .model import Base as Base
+from .model import BaseConfig as BaseConfig
+from .model import ConfigList as ConfigList
+from .model import LightningModuleBase as LightningModuleBase
+from .model import MetricConfig as MetricConfig
+from .runner import Runner as Runner
+from .trainer import Trainer as Trainer

{nshtrainer-0.8.2 → nshtrainer-0.8.4}/src/nshtrainer/ll/__init__.py

@@ -1,6 +1,7 @@
 from . import _experimental as _experimental
 from . import actsave as actsave
 from . import callbacks as callbacks
+from . import data as data
 from . import lr_scheduler as lr_scheduler
 from . import model as model
 from . import nn as nn

nshtrainer-0.8.4/src/nshtrainer/ll/_experimental.py

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

nshtrainer-0.8.4/src/nshtrainer/ll/snapshot.py

@@ -0,0 +1 @@
+from nshrunner.snapshot import *  # type: ignore # noqa: F403

nshtrainer-0.8.2/src/nshtrainer/__init__.py

@@ -1,42 +0,0 @@
-from . import _experimental as _experimental
-from . import callbacks as callbacks
-from . import lr_scheduler as lr_scheduler
-from . import model as model
-from . import nn as nn
-from . import optimizer as optimizer
-from .data import dataset_transform as dataset_transform
-from .lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
-from .model import ActSaveConfig as ActSaveConfig
-from .model import Base as Base
-from .model import BaseConfig as BaseConfig
-from .model import BaseLoggerConfig as BaseLoggerConfig
-from .model import BaseProfilerConfig as BaseProfilerConfig
-from .model import CheckpointLoadingConfig as CheckpointLoadingConfig
-from .model import CheckpointSavingConfig as CheckpointSavingConfig
-from .model import ConfigList as ConfigList
-from .model import DirectoryConfig as DirectoryConfig
-from .model import (
-    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
-)
-from .model import EnvironmentConfig as EnvironmentConfig
-from .model import (
-    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
-)
-from .model import (
-    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
-)
-from .model import GradientClippingConfig as GradientClippingConfig
-from .model import LightningModuleBase as LightningModuleBase
-from .model import LoggingConfig as LoggingConfig
-from .model import MetricConfig as MetricConfig
-from .model import OptimizationConfig as OptimizationConfig
-from .model import PrimaryMetricConfig as PrimaryMetricConfig
-from .model import ReproducibilityConfig as ReproducibilityConfig
-from .model import SanityCheckingConfig as SanityCheckingConfig
-from .model import TrainerConfig as TrainerConfig
-from .model import WandbWatchConfig as WandbWatchConfig
-from .nn import TypedModuleDict as TypedModuleDict
-from .nn import TypedModuleList as TypedModuleList
-from .optimizer import OptimizerConfig as OptimizerConfig
-from .runner import Runner as Runner
-from .trainer import Trainer as Trainer