nshtrainer 1.0.0b32__tar.gz → 1.0.0b33__tar.gz

{nshtrainer-1.0.0b32 → nshtrainer-1.0.0b33}/PKG-INFO

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

{nshtrainer-1.0.0b32 → nshtrainer-1.0.0b33}/pyproject.toml

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

{nshtrainer-1.0.0b32 → nshtrainer-1.0.0b33}/src/nshtrainer/data/datamodule.py

@@ -12,13 +12,11 @@ from typing_extensions import Never, TypeVar, deprecated, override
 
 from ..model.mixins.callback import CallbackRegistrarModuleMixin
 from ..model.mixins.debug import _DebugModuleMixin
-from ..util.hparams import HyperparamsMixin
 
 THparams = TypeVar("THparams", bound=C.Config, infer_variance=True)
 
 
 class LightningDataModuleBase(
-    HyperparamsMixin,
     _DebugModuleMixin,
     CallbackRegistrarModuleMixin,
     LightningDataModule,

{nshtrainer-1.0.0b32 → nshtrainer-1.0.0b33}/src/nshtrainer/model/base.py

@@ -16,7 +16,6 @@ from lightning.pytorch.utilities.rank_zero import rank_zero_warn
 from typing_extensions import Never, TypeVar, deprecated, override
 
 from ..callbacks.rlp_sanity_checks import _RLPSanityCheckModuleMixin
-from ..util.hparams import HyperparamsMixin
 from .mixins.callback import CallbackModuleMixin
 from .mixins.debug import _DebugModuleMixin
 from .mixins.logger import LoggerLightningModuleMixin
@@ -55,7 +54,6 @@ VALID_REDUCE_OPS = (
 
 
 class LightningModuleBase(
-    HyperparamsMixin,
     _DebugModuleMixin,
     _RLPSanityCheckModuleMixin,
     LoggerLightningModuleMixin,

nshtrainer-1.0.0b32/src/nshtrainer/util/hparams.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import nshconfig as C
-from lightning.pytorch.core.mixins.hparams_mixin import (
-    HyperparametersMixin as _LightningHyperparametersMixin,
-)
-
-
-class HyperparamsMixin(_LightningHyperparametersMixin):
-    if not TYPE_CHECKING:
-
-        def _to_hparams_dict(self, hp):
-            if isinstance(hp, C.Config):
-                return hp.model_dump(mode="python")
-
-            return super()._set_hparams(hp)