diff --git a/container_registry/container_analysis/samples_test.py b/container_registry/container_analysis/samples_test.py index 116796c0bf67..a8552e54742b 100644 --- a/container_registry/container_analysis/samples_test.py +++ b/container_registry/container_analysis/samples_test.py @@ -29,6 +29,7 @@ from grafeas.grafeas_v1.gapic.enums import NoteKind from grafeas.grafeas_v1.gapic.enums import Severity from grafeas.grafeas_v1.gapic.enums import Version +import pytest import samples @@ -145,6 +146,7 @@ def test_occurrences_for_note(self): # clean up samples.delete_occurrence(basename(occ.name), PROJECT_ID) + @pytest.mark.flaky(max_runs=3, min_passes=1) def test_pubsub(self): # create topic if needed client = SubscriberClient()