From 253579ac55843697e9368cb6bf6d8be271b4d700 Mon Sep 17 00:00:00 2001
From: nayaknishant <nishantnayak@google.com>
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/automl/snippets/src/main/java/com/example/automl/TranslateCreateModel.java b/automl/snippets/src/main/java/com/example/automl/TranslateCreateModel.java
index 8109e11a3cd..cddd597c6bb 100644
--- a/automl/snippets/src/main/java/com/example/automl/TranslateCreateModel.java
+++ b/automl/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/automl/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java b/automl/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java
index 71a3b7b0dee..65ce99331c7 100644
--- a/automl/snippets/src/test/java/com/example/automl/TranslateCreateModelTest.java
+++ b/automl/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");
}
}