Fixing tests

AI-SDC · Jul 26, 2023 · 3a4e35b · 3a4e35b
1 parent 96669b5
commit 3a4e35b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_safekeras2.py b/tests/test_safekeras2.py
@@ -330,11 +330,11 @@ def test_checkpoints_are_equal():
 
     same, msg = safekeras.check_checkpoint_equality("hello", "fit2.tf")
     assert same is False
-    assert "error_reloading_model_v1" in msg
+    assert "Error re-loading  model from" in msg
 
     same, msg = safekeras.check_checkpoint_equality("fit2.tf", "hello")
     assert same is False
-    assert "error_reloading_model_v2" in msg
+    assert "Error re-loading  model from" in msg
 
     for name in ("fit.tf", "fit2.tf", "refit.tf"):
         cleanup_file(name)