Skip to content
This repository was archived by the owner on Sep 5, 2023. It is now read-only.

Commit 4a01785

Browse files
author
Takashi Matsuo
authored
[datalabeling] testing: retry upon ServerError [(#3762)](GoogleCloudPlatform/python-docs-samples#3762)
fixes #3760
1 parent 4706e24 commit 4a01785

7 files changed

+17
-17
lines changed

samples/snippets/create_annotation_spec_set_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import create_annotation_spec_set
@@ -40,7 +40,7 @@ def cleaner():
4040
def test_create_annotation_spec_set(cleaner, capsys):
4141

4242
@backoff.on_exception(
43-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
43+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
4444
def run_sample():
4545
return create_annotation_spec_set.create_annotation_spec_set(PROJECT_ID)
4646

samples/snippets/create_instruction_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import create_instruction
@@ -42,7 +42,7 @@ def cleaner():
4242
def test_create_instruction(cleaner, capsys):
4343

4444
@backoff.on_exception(
45-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
45+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
4646
def run_sample():
4747
return create_instruction.create_instruction(
4848
PROJECT_ID, 'IMAGE', INSTRUCTION_GCS_URI)

samples/snippets/import_data_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import import_data
@@ -42,7 +42,7 @@ def dataset():
4242
def test_import_data(capsys, dataset):
4343

4444
@backoff.on_exception(
45-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
45+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
4646
def run_sample():
4747
import_data.import_data(dataset.name, 'IMAGE', INPUT_GCS_URI)
4848

samples/snippets/label_image_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import label_image
@@ -80,7 +80,7 @@ def test_label_image(
8080
capsys, annotation_spec_set, instruction, dataset, cleaner):
8181

8282
@backoff.on_exception(
83-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
83+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
8484
def run_sample():
8585
# Start labeling.
8686
return label_image.label_image(

samples/snippets/label_text_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import label_text
@@ -80,7 +80,7 @@ def cleaner():
8080
def test_label_text(capsys, annotation_spec_set, instruction, dataset, cleaner):
8181

8282
@backoff.on_exception(
83-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
83+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
8484
def run_sample():
8585
# Start labeling.
8686
return label_text.label_text(

samples/snippets/label_video_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
20+
from google.api_core.exceptions import ServerError
2121
import pytest
2222

2323
import label_video
@@ -80,7 +80,7 @@ def test_label_video(
8080
capsys, annotation_spec_set, instruction, dataset, cleaner):
8181

8282
@backoff.on_exception(
83-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
83+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
8484
def run_sample():
8585
# Start labeling.
8686
return label_video.label_video(

samples/snippets/manage_dataset_test.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
import os
1818

1919
import backoff
20-
from google.api_core.exceptions import DeadlineExceeded
2120
from google.api_core.exceptions import RetryError
21+
from google.api_core.exceptions import ServerError
2222
import pytest
2323

2424
import manage_dataset
@@ -60,7 +60,7 @@ def cleaner():
6060
def test_create_dataset(cleaner, capsys):
6161

6262
@backoff.on_exception(
63-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
63+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
6464
def run_sample():
6565
return manage_dataset.create_dataset(PROJECT_ID)
6666

@@ -75,7 +75,7 @@ def run_sample():
7575
def test_list_dataset(capsys, dataset):
7676

7777
@backoff.on_exception(
78-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
78+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
7979
def run_sample():
8080
manage_dataset.list_datasets(PROJECT_ID)
8181

@@ -86,7 +86,7 @@ def run_sample():
8686

8787
def test_get_dataset(capsys, dataset):
8888
@backoff.on_exception(
89-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
89+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
9090
def run_sample():
9191
manage_dataset.get_dataset(dataset.name)
9292

@@ -97,7 +97,7 @@ def run_sample():
9797

9898
def test_delete_dataset(capsys, dataset):
9999
@backoff.on_exception(
100-
backoff.expo, DeadlineExceeded, max_time=testing_lib.RETRY_DEADLINE)
100+
backoff.expo, ServerError, max_time=testing_lib.RETRY_DEADLINE)
101101
def run_sample():
102102
manage_dataset.delete_dataset(dataset.name)
103103

0 commit comments

Comments
 (0)