PolyTorch 0.1.3__tar.gz → 0.1.4__tar.gz

{polytorch-0.1.3 → polytorch-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: PolyTorch
-Version: 0.1.3
+Version: 0.1.4
 Summary: Embeddings and loss functions for different data types.
 License: Apache-2.0
 Keywords: pytorch,deep learning,command-line interface

{polytorch-0.1.3 → polytorch-0.1.4}/polytorch/data.py

@@ -12,7 +12,19 @@ from .enums import ContinuousLossType, BinaryLossType, CategoricalLossType
 @define(kw_only=True)
 class PolyData(abc.ABC):
     name: str = Factory(lambda self: self.__class__.__name__, takes_self=True)
-    loss_weighting: float = 1.0
+    loss_weighting: float = field(default=1.0)
+
+    def __setstate__(self, state):
+        # Handle (super_state, our_state) tuple pattern
+        if isinstance(state, tuple) and len(state) == 2:
+            _, actual_state = state
+        else:
+            actual_state = state
+
+        for k, v in actual_state.items():
+            setattr(self, k, v)
+        if not hasattr(self, "loss_weighting"):
+            self.loss_weighting = 1.0
 
     @abc.abstractmethod
     def embedding_module(self, embedding_size:int) -> nn.Module:
@@ -67,6 +79,9 @@ class BinaryData(PolyData):
         
         raise NotImplementedError(f"Unknown loss type: {self.loss_type} for {self.__class__.__name__}")
 
+    def __setstate__(self, state):
+        super().__setstate__(state)
+
 
 @define
 class CategoricalData(PolyData):
@@ -101,6 +116,9 @@ class CategoricalData(PolyData):
         
         raise NotImplementedError(f"Unknown loss type: {self.loss_type} for {self.__class__.__name__}")
 
+    def __setstate__(self, state):
+        super().__setstate__(state)
+        
 
 @define
 class OrdinalData(CategoricalData):
@@ -111,6 +129,9 @@ class OrdinalData(CategoricalData):
         from .embedding import OrdinalEmbedding
         return OrdinalEmbedding(self.category_count, embedding_size)
 
+    def __setstate__(self, state):
+        super().__setstate__(state)
+
 
 @define
 class ContinuousData(PolyData):
@@ -138,6 +159,9 @@ class ContinuousData(PolyData):
     def loss_func(self):
         return getattr(F, self.loss_type.name.lower())
 
+    def __setstate__(self, state):
+        super().__setstate__(state)
+
 
 @define
 class HierarchicalData(PolyData):
@@ -158,3 +182,5 @@ class HierarchicalData(PolyData):
     def loss_module(self):
         return HierarchicalSoftmaxLoss(root=self.root)
 
+    def __setstate__(self, state):
+        super().__setstate__(state)

{polytorch-0.1.3 → polytorch-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PolyTorch"
-version = "0.1.3"
+version = "0.1.4"
 description = "Embeddings and loss functions for different data types."
 authors = ["Robert Turnbull <robert.turnbull@unimelb.edu.au>"]
 license = "Apache-2.0"