From 9959aea2e3fca2ec2202a0b3cccb5a2aae793c72 Mon Sep 17 00:00:00 2001 From: nayaknishant Date: Fri, 10 Dec 2021 10:40:17 -0800 Subject: [PATCH] docs: removed default model comment and added comment to test (#997) --- .../main/java/com/example/automl/TranslateCreateModel.java | 1 - .../java/com/example/automl/TranslateCreateModelTest.java | 4 ++++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/samples/snippets/src/main/java/com/example/automl/TranslateCreateModel.java b/samples/snippets/src/main/java/com/example/automl/TranslateCreateModel.java index 8109e11a3..cddd597c6 100644 --- a/samples/snippets/src/main/java/com/example/automl/TranslateCreateModel.java +++ b/samples/snippets/src/main/java/com/example/automl/TranslateCreateModel.java @@ -46,7 +46,6 @@ static void createModel(String projectId, String datasetId, String displayName) try (AutoMlClient client = AutoMlClient.create()) { // A resource that represents Google Cloud Platform location. LocationName projectLocation = LocationName.of(projectId, "us-central1"); - // Leave model unset to use the default base model provided by Google TranslationModelMetadata translationModelMetadata = TranslationModelMetadata.newBuilder().build(); Model model = diff --git a/samples/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java b/samples/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java index 71a3b7b0d..65ce99331 100644 --- a/samples/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java +++ b/samples/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java @@ -76,8 +76,12 @@ public void testTranslateCreateModel() { String.format("test_%s", UUID.randomUUID().toString().replace("-", "_").substring(0, 26)); TranslateCreateModel.createModel(PROJECT_ID, DATASET_ID, modelName); String got = bout.toString(); + // After setting DATASET_ID, change line below to + // assertThat(got).contains("Training started..."); assertThat(got).contains("Dataset does not exist"); } catch (IOException | ExecutionException | InterruptedException e) { + // After setting DATASET_ID, change line below to + // assertThat(e.getMessage()).contains("Training started..."); assertThat(e.getMessage()).contains("Dataset does not exist"); } }