Skip to content

Commit 066bac1

Browse files
authored
chore: remove unneeded Translation and NL references (#129)
* chore: remove unneeded Translation and NL references * fix: tests
1 parent c65b7b9 commit 066bac1

File tree

7 files changed

+5
-368
lines changed

7 files changed

+5
-368
lines changed

automl/beta/delete_dataset_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ def dataset_id():
2929
client = automl.AutoMlClient()
3030
project_location = f"projects/{PROJECT_ID}/locations/us-central1"
3131
display_name = "test_{}".format(uuid.uuid4()).replace("-", "")[:32]
32-
metadata = automl.TextExtractionDatasetMetadata()
32+
metadata = automl.VideoClassificationDatasetMetadata()
3333
dataset = automl.Dataset(
34-
display_name=display_name, text_extraction_dataset_metadata=metadata
34+
display_name=display_name, video_classification_dataset_metadata=metadata
3535
)
3636
response = client.create_dataset(parent=project_location, dataset=dataset)
3737
dataset_id = response.name.split("/")[-1]

automl/beta/delete_model_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_delete_model(capsys):
2424
# nonexistent model and confirm that the model was not found, but other
2525
# elements of the request were valid.
2626
try:
27-
delete_model.delete_model(PROJECT_ID, "TRL0000000000000000000")
27+
delete_model.delete_model(PROJECT_ID, "VCN0000000000000000000")
2828
out, _ = capsys.readouterr()
2929
assert "The model does not exist" in out
3030
except Exception as e:

automl/beta/import_dataset_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
PROJECT_ID = os.environ["AUTOML_PROJECT_ID"]
2020
BUCKET_ID = "{}-lcm".format(PROJECT_ID)
21-
DATASET_ID = "TEN0000000000000000000"
21+
DATASET_ID = "VCN0000000000000000000"
2222

2323

2424
def test_import_dataset(capsys):
@@ -27,7 +27,7 @@ def test_import_dataset(capsys):
2727
# confirm that the dataset was not found, but other elements of the request
2828
# were valid.
2929
try:
30-
data = "gs://{}/sentiment-analysis/dataset.csv".format(BUCKET_ID)
30+
data = "gs://{}/video-classification/dataset.csv".format(BUCKET_ID)
3131
import_dataset.import_dataset(PROJECT_ID, DATASET_ID, data)
3232
out, _ = capsys.readouterr()
3333
assert (

automl/snippets/automl_translation_dataset.py

Lines changed: 0 additions & 106 deletions
This file was deleted.

automl/snippets/automl_translation_model.py

Lines changed: 0 additions & 176 deletions
This file was deleted.

automl/snippets/dataset_test.py

Lines changed: 0 additions & 45 deletions
This file was deleted.

0 commit comments

Comments
 (0)