Skip to content

Commit 559b98d

Browse files
committed
fix unserializable link
1 parent a9a760a commit 559b98d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bayesflow/links/positive_semi_definite.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
import keras
2+
from keras.saving import register_keras_serializable as serializable
23

34
from bayesflow.utils import keras_kwargs
45

56

7+
@serializable(package="bayesflow.links")
68
class PositiveSemiDefinite(keras.Layer):
79
"""Activation function to link from any square matrix to a positive semidefinite matrix."""
810

0 commit comments

Comments
 (0)