Skip to content

Commit 41b70f4

Browse files
authored
Add Set Endpoint Samples (#2497)
* Add Set Endpoint Samples * Add additional test result option * Sample Request update * Add filter_
1 parent 7b13271 commit 41b70f4

File tree

6 files changed

+132
-2
lines changed

6 files changed

+132
-2
lines changed

automl/beta/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
google-cloud-automl==0.7.0

automl/beta/set_endpoint.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
# Copyright 2019 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
def set_endpoint(project_id):
17+
"""Change your endpoint"""
18+
# [START automl_set_endpoint]
19+
from google.cloud import automl_v1beta1 as automl
20+
21+
# You must first create a dataset, using the `eu` endpoint, before you can
22+
# call other operations such as: list, get, import, delete, etc.
23+
client_options = {'api_endpoint': 'eu-automl.googleapis.com:443'}
24+
25+
# Instantiates a client
26+
client = automl.AutoMlClient(client_options=client_options)
27+
28+
# A resource that represents Google Cloud Platform location.
29+
# project_id = 'YOUR_PROJECT_ID'
30+
project_location = client.location_path(project_id, 'eu')
31+
# [END automl_set_endpoint]
32+
33+
# List all the datasets available
34+
# Note: Create a dataset in `eu`, before calling `list_datasets`.
35+
response = client.list_datasets(
36+
project_location, filter_='')
37+
38+
for dataset in response:
39+
print(dataset)

automl/beta/set_endpoint_test.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
# Copyright 2019 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
import os
16+
import set_endpoint
17+
18+
PROJECT_ID = os.environ['GCLOUD_PROJECT']
19+
20+
21+
def test_set_endpoint(capsys):
22+
set_endpoint.set_endpoint(PROJECT_ID)
23+
24+
out, _ = capsys.readouterr()
25+
# Look for the display name
26+
assert 'do_not_delete_me' in out
Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
# Copyright 2019 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
def set_endpoint():
17+
"""Change your endpoint"""
18+
# [START language_set_endpoint]
19+
# Imports the Google Cloud client library
20+
from google.cloud import language
21+
22+
client_options = {'api_endpoint': 'eu-language.googleapis.com:443'}
23+
24+
# Instantiates a client
25+
client = language.LanguageServiceClient(client_options=client_options)
26+
# [END language_set_endpoint]
27+
28+
# The text to analyze
29+
document = language.types.Document(
30+
content='Hello, world!',
31+
type=language.enums.Document.Type.PLAIN_TEXT)
32+
33+
# Detects the sentiment of the text
34+
sentiment = client.analyze_sentiment(document=document).document_sentiment
35+
36+
print('Sentiment: {}, {}'.format(sentiment.score, sentiment.magnitude))
37+
38+
39+
if __name__ == '__main__':
40+
set_endpoint()
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
# Copyright 2019 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
import set_endpoint
16+
17+
18+
def test_set_endpoint(capsys):
19+
set_endpoint.set_endpoint()
20+
21+
out, _ = capsys.readouterr()
22+
assert 'Sentiment' in out

vision/cloud-client/detect/detect_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,15 +225,17 @@ def test_detect_web_with_geo(capsys):
225225
detect.web_entities_include_geo_results(file_name)
226226
out, _ = capsys.readouterr()
227227
out = out.lower()
228-
assert 'zepra' in out or 'electra tower' in out or 'tel aviv' in out
228+
assert ('zepra' in out or 'electra tower' in out or 'tel aviv' in out or
229+
'jaffa' in out)
229230

230231

231232
def test_detect_web_with_geo_uri(capsys):
232233
file_name = 'gs://{}/vision/web/city.jpg'.format(ASSET_BUCKET)
233234
detect.web_entities_include_geo_results_uri(file_name)
234235
out, _ = capsys.readouterr()
235236
out = out.lower()
236-
assert 'zepra' in out or 'electra tower' in out or 'tel aviv' in out
237+
assert ('zepra' in out or 'electra tower' in out or 'tel aviv' in out or
238+
'jaffa' in out)
237239

238240

239241
def test_detect_document(capsys):

0 commit comments

Comments
 (0)