Increase test coverage in saving

keras-team · Sep 16, 2023 · 4f51acf · 4f51acf
1 parent e0a76fa
commit 4f51acf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras_core/saving/saving_api_test.py b/keras_core/saving/saving_api_test.py
@@ -6,6 +6,7 @@
 from keras_core import layers
 from keras_core.models import Sequential
 from keras_core.saving import saving_api
+from keras_core.utils import io_utils
 
 
 class SaveModelTests(unittest.TestCase):
@@ -34,7 +35,7 @@ def test_invalid_save_format(self):
 
     def test_overwrite_prompt(self):
         original_mtime = os.path.getmtime(self.filepath)
-        saving_api.io_utils.ask_to_proceed_with_overwrite = lambda x: False
+        io_utils.ask_to_proceed_with_overwrite = lambda x: False
         saving_api.save_model(self.model, self.filepath, overwrite=False)
         new_mtime = os.path.getmtime(self.filepath)
         self.assertEqual(original_mtime, new_mtime)