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

GoogleCloudPlatform · Nov 18, 2022 · 9ac9c23 · 9ac9c23
1 parent cd35d2d
commit 9ac9c23
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");
     }
   }
 }