samples: automl: update error check due to changes (#2377)

GoogleCloudPlatform · Dec 5, 2022 · e77f16a · e77f16a
1 parent 3203f3f
commit e77f16a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/automl/snippets/src/test/java/com/example/automl/BatchPredictTest.java b/automl/snippets/src/test/java/com/example/automl/BatchPredictTest.java
@@ -83,10 +83,10 @@ public void testBatchPredict() {
       BatchPredict.batchPredict(PROJECT_ID, MODEL_ID, inputUri, outputUri);
       String got = bout.toString();
       assertThat(got)
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     } catch (IOException | ExecutionException | InterruptedException e) {
       assertThat(e.getMessage())
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     }
   }
 }