Skip to content

datalabeling: ensure all tests use test endpoint #2918

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nnegrey merged 6 commits into from
Mar 18, 2020
Merged

datalabeling: ensure all tests use test endpoint #2918

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
58e56fc
datalabeling: ensure all tests use test endpoint
nnegrey Feb 14, 2020
cd04907
Merge branch 'master' into fix-datalabeling-tests
nnegrey Feb 19, 2020
1e4d487
Merge branch 'master' into fix-datalabeling-tests
nnegrey Feb 24, 2020
841c011
Merge branch 'master' into fix-datalabeling-tests
leahecole Mar 5, 2020
3970b7c
requires an input csv for text input, slight print statement cleanup
nnegrey Mar 18, 2020
c97284e
Merge branch 'master' into fix-datalabeling-tests
nnegrey Mar 18, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions datalabeling/manage_dataset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,13 @@ def list_datasets(project_id):
"""Lists datasets for the given Google Cloud project."""
from google.cloud import datalabeling_v1beta1 as datalabeling
client = datalabeling.DataLabelingServiceClient()
# [END datalabeling_list_datasets_beta]
# If provided, use a provided test endpoint - this will prevent tests on
# this snippet from triggering any action by a real human
if 'DATALABELING_ENDPOINT' in os.environ:
opts = ClientOptions(api_endpoint=os.getenv('DATALABELING_ENDPOINT'))
client = datalabeling.DataLabelingServiceClient(client_options=opts)
# [START datalabeling_list_datasets_beta]

formatted_project_name = client.project_path(project_id)

Expand All @@ -80,6 +87,13 @@ def get_dataset(dataset_resource_name):
"""Gets a dataset for the given Google Cloud project."""
from google.cloud import datalabeling_v1beta1 as datalabeling
client = datalabeling.DataLabelingServiceClient()
# [END datalabeling_get_dataset_beta]
# If provided, use a provided test endpoint - this will prevent tests on
# this snippet from triggering any action by a real human
if 'DATALABELING_ENDPOINT' in os.environ:
opts = ClientOptions(api_endpoint=os.getenv('DATALABELING_ENDPOINT'))
client = datalabeling.DataLabelingServiceClient(client_options=opts)
# [START datalabeling_get_dataset_beta]

response = client.get_dataset(dataset_resource_name)

Expand All @@ -97,6 +111,13 @@ def delete_dataset(dataset_resource_name):
"""Deletes a dataset for the given Google Cloud project."""
from google.cloud import datalabeling_v1beta1 as datalabeling
client = datalabeling.DataLabelingServiceClient()
# [END datalabeling_delete_dataset_beta]
# If provided, use a provided test endpoint - this will prevent tests on
# this snippet from triggering any action by a real human
if 'DATALABELING_ENDPOINT' in os.environ:
opts = ClientOptions(api_endpoint=os.getenv('DATALABELING_ENDPOINT'))
client = datalabeling.DataLabelingServiceClient(client_options=opts)
# [START datalabeling_delete_dataset_beta]

response = client.delete_dataset(dataset_resource_name)

Expand Down