cannot suppport filtering kwargs for numpy ufunc in python 3.10

Author: LarsKue

bayesflow/adapters/transforms/expand_dims.py

@@ -1,9 +1,9 @@
 import numpy as np
 from keras.saving import (
     deserialize_keras_object as deserialize,
+    register_keras_serializable as serializable,
     serialize_keras_object as serialize,
 )
-from keras.saving import register_keras_serializable as serializable
 
 from .elementwise_transform import ElementwiseTransform
 

bayesflow/adapters/transforms/log.py

@@ -1,9 +1,9 @@
 import numpy as np
 from keras.saving import (
     deserialize_keras_object as deserialize,
+    register_keras_serializable as serializable,
     serialize_keras_object as serialize,
 )
-from keras.saving import register_keras_serializable as serializable
 
 from .elementwise_transform import ElementwiseTransform
 

bayesflow/adapters/transforms/numpy_transform.py

@@ -1,7 +1,6 @@
 import numpy as np
 from keras.saving import register_keras_serializable as serializable
 
-from bayesflow.utils import filter_kwargs
 from .elementwise_transform import ElementwiseTransform
 
 
@@ -74,9 +73,7 @@ def get_config(self) -> dict:
         return {"forward": self._forward.__name__, "inverse": self._inverse.__name__}
 
     def forward(self, data: dict[str, any], **kwargs) -> dict[str, any]:
-        kwargs = filter_kwargs(kwargs, self._forward)
-        return self._forward(data, **kwargs)
+        return self._forward(data)
 
     def inverse(self, data: np.ndarray, **kwargs) -> np.ndarray:
-        kwargs = filter_kwargs(kwargs, self._inverse)
-        return self._inverse(data, **kwargs)
+        return self._inverse(data)

bayesflow/adapters/transforms/to_array.py

@@ -1,9 +1,7 @@
 from numbers import Number
 
 import numpy as np
-from keras.saving import (
-    register_keras_serializable as serializable,
-)
+from keras.saving import register_keras_serializable as serializable
 
 from bayesflow.utils.io import deserialize_type, serialize_type
 from .elementwise_transform import ElementwiseTransform