From e77f16ac4e7c321f1a6b70a02c835cb6d7739352 Mon Sep 17 00:00:00 2001 From: Noah Negrey Date: Wed, 11 Mar 2020 11:03:46 -0600 Subject: [PATCH] samples: automl: update error check due to changes (#2377) --- .../src/test/java/com/example/automl/BatchPredictTest.java | 4 ++-- 1 file changed, 2 insertions(+), 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 index d46b669ac1d..974caaa6e7d 100644 --- 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"); } } }