automl: fix batch test due to error message change (#3057)

fix: #3052
GoogleCloudPlatform · Mar 6, 2020 · 1718111 · 1718111
1 parent 3bfad2c
commit 1718111
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/automl/cloud-client/batch_predict_test.py b/automl/cloud-client/batch_predict_test.py
@@ -37,11 +37,11 @@ def test_batch_predict(capsys):
         )
         out, _ = capsys.readouterr()
         assert (
-            "The model is either not found or not supported for prediction yet"
+            "does not exist"
             in out
         )
     except Exception as e:
         assert (
-            "The model is either not found or not supported for prediction yet"
+            "does not exist"
             in e.message
         )