fix: make transforms AsSet and AsTimeSeries serializable

closes #302
bayesflow-org · Feb 11, 2025 · 0a4921b · 0a4921b
1 parent 1081bbd
commit 0a4921b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/bayesflow/adapters/transforms/as_set.py b/bayesflow/adapters/transforms/as_set.py
@@ -33,3 +33,10 @@ def inverse(self, data: np.ndarray, **kwargs) -> np.ndarray:
             return np.squeeze(data, axis=2)
 
         return data
+
+    @classmethod
+    def from_config(cls, config: dict, custom_objects=None) -> "AsSet":
+        return cls()
+
+    def get_config(self) -> dict:
+        return {}
diff --git a/bayesflow/adapters/transforms/as_time_series.py b/bayesflow/adapters/transforms/as_time_series.py
@@ -30,3 +30,10 @@ def inverse(self, data: np.ndarray, **kwargs) -> np.ndarray:
             return np.squeeze(data, axis=2)
 
         return data
+
+    @classmethod
+    def from_config(cls, config: dict, custom_objects=None) -> "AsTimeSeries":
+        return cls()
+
+    def get_config(self) -> dict:
+        return {}