Update keras.saving for legacy

Update keras.saving for the packages having moved to legacy
TileDB-Inc · Jul 17, 2023 · 142c6e8 · 142c6e8
1 parent d763940
commit 142c6e8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tiledb/ml/models/tensorflow_keras.py b/tiledb/ml/models/tensorflow_keras.py
@@ -18,9 +18,11 @@
 SharedObjectLoadingScope = keras.utils.generic_utils.SharedObjectLoadingScope
 FunctionalOrSequential = (keras.models.Functional, keras.models.Sequential)
 TFOptimizer = keras.optimizers.TFOptimizer
-get_json_type = keras.saving.saved_model.json_utils.get_json_type
-preprocess_weights_for_loading = keras.saving.hdf5_format.preprocess_weights_for_loading
-saving_utils = keras.saving.saving_utils
+get_json_type = keras.saving.legacy.saved_model.json_utils.get_json_type
+preprocess_weights_for_loading = (
+    keras.saving.legacy.hdf5_format.preprocess_weights_for_loading
+)
+saving_utils = keras.saving.legacy.saving_utils
 
 
 class TensorflowKerasTileDBModel(TileDBArtifact[tf.keras.Model]):