Skip to content

Commit

Permalink
Add Set Endpoint Samples [(#2497)](#2497)
Browse files Browse the repository at this point in the history
* Add Set Endpoint Samples

* Add additional test result option

* Sample Request update

* Add filter_
  • Loading branch information
nnegrey authored and busunkim96 committed Aug 13, 2020
1 parent c9cd8ed commit 2f16326
Show file tree
Hide file tree
Showing 3 changed files with 66 additions and 0 deletions.
1 change: 1 addition & 0 deletions automl/beta/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
google-cloud-automl==0.7.0
39 changes: 39 additions & 0 deletions automl/beta/set_endpoint.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
# Copyright 2019 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


def set_endpoint(project_id):
"""Change your endpoint"""
# [START automl_set_endpoint]
from google.cloud import automl_v1beta1 as automl

# You must first create a dataset, using the `eu` endpoint, before you can
# call other operations such as: list, get, import, delete, etc.
client_options = {'api_endpoint': 'eu-automl.googleapis.com:443'}

# Instantiates a client
client = automl.AutoMlClient(client_options=client_options)

# A resource that represents Google Cloud Platform location.
# project_id = 'YOUR_PROJECT_ID'
project_location = client.location_path(project_id, 'eu')
# [END automl_set_endpoint]

# List all the datasets available
# Note: Create a dataset in `eu`, before calling `list_datasets`.
response = client.list_datasets(
project_location, filter_='')

for dataset in response:
print(dataset)
26 changes: 26 additions & 0 deletions automl/beta/set_endpoint_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
# Copyright 2019 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import set_endpoint

PROJECT_ID = os.environ['GCLOUD_PROJECT']


def test_set_endpoint(capsys):
set_endpoint.set_endpoint(PROJECT_ID)

out, _ = capsys.readouterr()
# Look for the display name
assert 'do_not_delete_me' in out

0 comments on commit 2f16326

Please sign in to comment.