Fix utils namespace

mikbuch · Mar 25, 2023 · 5a95a6d · 5a95a6d
1 parent 3b39310
commit 5a95a6d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/saving_master.py b/scripts/saving_master.py
@@ -42,8 +42,8 @@
             "path": "serialization_utils",
             "title": "Serialization utilities",
             "generate": [
-                "tensorflow.keras.saving.serialize_keras_object",
-                "tensorflow.keras.saving.deserialize_keras_object",
+                "tensorflow.keras.utils.serialize_keras_object",  # TODO: move to saving
+                "tensorflow.keras.utils.deserialize_keras_object",  # TODO: move to saving
                 "tensorflow.keras.saving.custom_object_scope",
                 "tensorflow.keras.saving.get_custom_objects",
                 "tensorflow.keras.saving.register_keras_serializable",