chore: refactor translation automl test (#113)

Fixes #111 🦕
GoogleCloudPlatform · Dec 22, 2020 · 660077a · 660077a
1 parent d98a59a
commit 660077a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automl/snippets/dataset_test.py b/automl/snippets/dataset_test.py
@@ -23,6 +23,7 @@
 
 project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
 compute_region = "us-central1"
+dataset_id = "TRL3876092572857648864"
 
 
 @pytest.mark.slow
@@ -61,7 +62,6 @@ def test_dataset_list_get(capsys):
     assert "Dataset id: " in list_dataset_output[2]
 
     # get dataset
-    dataset_id = list_dataset_output[2].split()[2]
     automl_translation_dataset.get_dataset(
         project_id, compute_region, dataset_id
     )