experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Revert "add save load test for pathlib path" 

This reverts commit 1363f0f6e8.
This commit is contained in:
Yixing Fu 2020-06-18 20:38:24 -04:00
parent 9f0e739f5f
commit 5869f65762
1 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/python/keras/saving/save_test.py
View File

@ -71,12 +71,6 @@ class TestSaveModel(test.TestCase, parameterized.TestCase):
save.save_model(self.model, path) save.save_model(self.model, path)
self.assert_saved_model(path) self.assert_saved_model(path)
@test_util.run_v2_only
def test_save_format_defaults_pathlib(self):
path = pathlib.Path(self.get_temp_dir()) / 'model_path'
save.save_model(self.model, path)
self.assert_saved_model(path)
@test_util.run_v2_only @test_util.run_v2_only
def test_save_hdf5(self): def test_save_hdf5(self):
path = os.path.join(self.get_temp_dir(), 'model') path = os.path.join(self.get_temp_dir(), 'model')
@ -87,13 +81,6 @@ class TestSaveModel(test.TestCase, parameterized.TestCase):
'requires the model to be a Functional model or a Sequential model.'): 'requires the model to be a Functional model or a Sequential model.'):
save.save_model(self.subclassed_model, path, save_format='h5') save.save_model(self.subclassed_model, path, save_format='h5')
@test_util.run_v2_only
def test_save_load_hdf5_pathlib(self):
if sys.version_info >= (3, 6):
path = pathlib.Path(self.get_temp_dir()) / 'model'
save.save_model(self.model, path, save_format='h5')
save.load_model(path)
@test_util.run_v2_only @test_util.run_v2_only
def test_save_tf(self): def test_save_tf(self):
path = os.path.join(self.get_temp_dir(), 'model') path = os.path.join(self.get_temp_dir(), 'model')