nshtrainer 1.0.0b30__tar.gz → 1.0.0b31__tar.gz

{nshtrainer-1.0.0b30 → nshtrainer-1.0.0b31}/PKG-INFO

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

{nshtrainer-1.0.0b30 → nshtrainer-1.0.0b31}/pyproject.toml

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

{nshtrainer-1.0.0b30 → nshtrainer-1.0.0b31}/src/nshtrainer/data/datamodule.py

@@ -12,11 +12,13 @@ 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.0b30 → nshtrainer-1.0.0b31}/src/nshtrainer/model/base.py

@@ -16,6 +16,7 @@ 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
@@ -54,6 +55,7 @@ VALID_REDUCE_OPS = (
 
 
 class LightningModuleBase(
+    HyperparamsMixin,
     _DebugModuleMixin,
     _RLPSanityCheckModuleMixin,
     LoggerLightningModuleMixin,

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

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+import nshconfig as C
+from lightning.pytorch.core.mixins.hparams_mixin import (
+    HyperparametersMixin as _LightningHyperparametersMixin,
+)
+from typing_extensions import override
+
+
+class HyperparamsMixin(_LightningHyperparametersMixin):
+    @override
+    def _set_hparams(self, hp):
+        if isinstance(hp, C.Config):
+            self._hparams = hp
+            return
+
+        return super()._set_hparams(hp)