fix engine training tests ValueError message on empty input

keras-team · Apr 27, 2023 · 8c0bd19 · 8c0bd19
1 parent 0a66368
commit 8c0bd19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras/engine/training_test.py b/keras/engine/training_test.py
@@ -94,7 +94,7 @@ def test_fit_on_empty(self):
         model = sequential.Sequential([layers_module.Dense(1)])
         model.compile("sgd", "mse", run_eagerly=test_utils.should_run_eagerly())
         with self.assertRaisesRegex(
-            ValueError, "Unexpected result of `train_function`.*"
+            ValueError, "Expected input data to be non-empty."
         ):
             model.fit(x=np.array([]), y=np.array([]))
 
@@ -2448,7 +2448,7 @@ def test_predict_error_with_empty_x(self):
         model.compile(loss="mse")
 
         with self.assertRaisesRegex(
-            ValueError, "Unexpected result of `predict_function`.*"
+            ValueError, "Expected input data to be non-empty."
         ):
             model.predict(np.array([]))