nshtrainer 1.0.0b31__tar.gz → 1.0.0b32__tar.gz

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

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

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

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

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

@@ -0,0 +1,18 @@
+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)

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

@@ -1,17 +0,0 @@
-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)