chore: remove unneeded Translation and NL references (#129)

* chore: remove unneeded Translation and NL references * fix: tests
GoogleCloudPlatform · Nov 17, 2022 · 7ab804d · 7ab804d
1 parent bacd7d3
commit 7ab804d
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 368 deletions.
diff --git a/automl/beta/delete_dataset_test.py b/automl/beta/delete_dataset_test.py
@@ -29,9 +29,9 @@ def dataset_id():
     client = automl.AutoMlClient()
     project_location = f"projects/{PROJECT_ID}/locations/us-central1"
     display_name = "test_{}".format(uuid.uuid4()).replace("-", "")[:32]
-    metadata = automl.TextExtractionDatasetMetadata()
+    metadata = automl.VideoClassificationDatasetMetadata()
     dataset = automl.Dataset(
-        display_name=display_name, text_extraction_dataset_metadata=metadata
+        display_name=display_name, video_classification_dataset_metadata=metadata
     )
     response = client.create_dataset(parent=project_location, dataset=dataset)
     dataset_id = response.name.split("/")[-1]

diff --git a/automl/beta/delete_model_test.py b/automl/beta/delete_model_test.py
@@ -24,7 +24,7 @@ def test_delete_model(capsys):
     # nonexistent model and confirm that the model was not found, but other
     # elements of the request were valid.
     try:
-        delete_model.delete_model(PROJECT_ID, "TRL0000000000000000000")
+        delete_model.delete_model(PROJECT_ID, "VCN0000000000000000000")
         out, _ = capsys.readouterr()
         assert "The model does not exist" in out
     except Exception as e:

diff --git a/automl/beta/import_dataset_test.py b/automl/beta/import_dataset_test.py
@@ -18,7 +18,7 @@
 
 PROJECT_ID = os.environ["AUTOML_PROJECT_ID"]
 BUCKET_ID = "{}-lcm".format(PROJECT_ID)
-DATASET_ID = "TEN0000000000000000000"
+DATASET_ID = "VCN0000000000000000000"
 
 
 def test_import_dataset(capsys):
@@ -27,7 +27,7 @@ def test_import_dataset(capsys):
     # confirm that the dataset was not found, but other elements of the request
     # were valid.
     try:
-        data = "gs://{}/sentiment-analysis/dataset.csv".format(BUCKET_ID)
+        data = "gs://{}/video-classification/dataset.csv".format(BUCKET_ID)
         import_dataset.import_dataset(PROJECT_ID, DATASET_ID, data)
         out, _ = capsys.readouterr()
         assert (

diff --git a/automl/snippets/automl_translation_dataset.py b/automl/snippets/automl_translation_dataset.py
diff --git a/automl/snippets/automl_translation_model.py b/automl/snippets/automl_translation_model.py
diff --git a/automl/snippets/dataset_test.py b/automl/snippets/dataset_test.py