Remove deprecated save model option.

linqs · May 8, 2024 · aadeb47 · aadeb47
1 parent 3838cd1
commit aadeb47
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/psl-python/tests/python/deeppsl/test_tensorflow_models.py b/psl-python/tests/python/deeppsl/test_tensorflow_models.py
@@ -69,7 +69,7 @@ def test_sign_model(self):
         post_train_results = sign_model.internal_eval(test_data, options=options)
 
         with tempfile.TemporaryDirectory(suffix = '_TestNeuPSL') as temp_dir:
-            save_path = os.path.join(temp_dir, 'tensorflow_model')
+            save_path = os.path.join(temp_dir, 'tensorflow_model.h5')
             options['save_path'] = save_path
             options['load_path'] = save_path
             sign_model.internal_save(options=options)

diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
@@ -71,7 +71,7 @@ def internal_save(self, options):
         if 'save_path' not in options:
             return {}
 
-        self._model.save(options['save_path'], save_format='tf')
+        self._model.save(options['save_path'])
         return {}
 
     def load(self, options):