diff --git a/sdks/python/apache_beam/examples/wordcount_it_test.py b/sdks/python/apache_beam/examples/wordcount_it_test.py index 0af89283ea92..2030986d60b5 100644 --- a/sdks/python/apache_beam/examples/wordcount_it_test.py +++ b/sdks/python/apache_beam/examples/wordcount_it_test.py @@ -47,7 +47,7 @@ class WordCountIT(unittest.TestCase): def test_wordcount_it(self): self._run_wordcount_it(wordcount.run) - @pytest.mark.it_postcommit + @pytest.mark.it_postcommit_impersonation @pytest.mark.sickbay_direct @pytest.mark.sickbay_spark @pytest.mark.sickbay_flink diff --git a/sdks/python/test-suites/dataflow/common.gradle b/sdks/python/test-suites/dataflow/common.gradle index fdc428001c3b..bfc672b00abf 100644 --- a/sdks/python/test-suites/dataflow/common.gradle +++ b/sdks/python/test-suites/dataflow/common.gradle @@ -115,7 +115,7 @@ task postCommitIT { "sdk_location": files(configurations.distTarBall.files).singleFile, "worker_jar": dataflowWorkerJar, "suite": "postCommitIT-df${pythonVersionSuffix}", - "collect": "it_postcommit" + "collect": "it_postcommit_impersonation" ] def cmdArgs = mapToArgString(argMap) exec {