fix adapter serialization

Author: LarsKue

bayesflow/adapters/transforms/as_set.py

@@ -1,8 +1,10 @@
+from keras.saving import register_keras_serializable as serializable
 import numpy as np
 
 from .elementwise_transform import ElementwiseTransform
 
 
+@serializable(package="bayesflow.adapters")
 class AsSet(ElementwiseTransform):
     """The `.as_set(["x", "y"])` transform indicates that both `x` and `y` are treated as sets.
 

bayesflow/adapters/transforms/expand_dims.py

@@ -1,3 +1,4 @@
+from keras.saving import register_keras_serializable as serializable
 import numpy as np
 
 from keras.saving import (
@@ -8,6 +9,7 @@
 from .elementwise_transform import ElementwiseTransform
 
 
+@serializable(package="bayesflow.adapters")
 class ExpandDims(ElementwiseTransform):
     """
     Expand the shape of an array.

bayesflow/adapters/transforms/log.py

@@ -1,3 +1,4 @@
+from keras.saving import register_keras_serializable as serializable
 import numpy as np
 
 from keras.saving import (
@@ -8,6 +9,7 @@
 from .elementwise_transform import ElementwiseTransform
 
 
+@serializable(package="bayesflow.adapters")
 class Log(ElementwiseTransform):
     """Log transforms a variable.
 

bayesflow/adapters/transforms/sqrt.py

@@ -1,8 +1,10 @@
+from keras.saving import register_keras_serializable as serializable
 import numpy as np
 
 from .elementwise_transform import ElementwiseTransform
 
 
+@serializable(package="bayesflow.adapters")
 class Sqrt(ElementwiseTransform):
     """Square-root transform a variable.