diff --git a/keras_hub/src/api_export.py b/keras_hub/src/api_export.py index 1cc8322093..22199959c1 100644 --- a/keras_hub/src/api_export.py +++ b/keras_hub/src/api_export.py @@ -52,8 +52,8 @@ def __call__(self, symbol): class keras_hub_export: def __init__(self, path): - pass + self.path = path def __call__(self, symbol): - maybe_register_serializable(symbol) + maybe_register_serializable(self.path, symbol) return symbol