Skip to content

Commit d500b4d

Browse files
authored
Fix Keras RS object serialization. (#17)
The `keras_rs_export` decorator was not calling `register_keras_serializable` correctly.
1 parent 5164294 commit d500b4d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

keras_rs/src/api_export.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,11 @@
1313

1414
class keras_rs_export:
1515
def __init__(self, path: str):
16-
self.path = path
1716
if namex is not None:
1817
self.namex_export = namex.export(package="keras_rs", path=path)
1918

2019
def __call__(self, symbol: T) -> T:
21-
keras.saving.register_keras_serializable(self.path, symbol)
20+
keras.saving.register_keras_serializable(package="keras_rs")(symbol)
2221
if namex is not None:
2322
self.namex_export(symbol)
2423
return symbol

0 commit comments

Comments
 (0)