Fix Keras RS object serialization. (#17)

The `keras_rs_export` decorator was not calling `register_keras_serializable` correctly.
keras-team · Jan 28, 2025 · d500b4d · d500b4d
1 parent 5164294
commit d500b4d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/keras_rs/src/api_export.py b/keras_rs/src/api_export.py
@@ -13,12 +13,11 @@
 
 class keras_rs_export:
     def __init__(self, path: str):
-        self.path = path
         if namex is not None:
             self.namex_export = namex.export(package="keras_rs", path=path)
 
     def __call__(self, symbol: T) -> T:
-        keras.saving.register_keras_serializable(self.path, symbol)
+        keras.saving.register_keras_serializable(package="keras_rs")(symbol)
         if namex is not None:
             self.namex_export(symbol)
         return symbol