added dictionary back

apache · May 12, 2022 · 1ec2b1d · 1ec2b1d
1 parent 11857a2
commit 1ec2b1d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/internal/gcp/auth.py b/sdks/python/apache_beam/internal/gcp/auth.py
@@ -173,6 +173,9 @@ def _add_impersonation_credentials(credentials, pipeline_options):
     if isinstance(pipeline_options, PipelineOptions):
       gcs_options = pipeline_options.view_as(GoogleCloudOptions)
       impersonate_service_account = gcs_options.impersonate_service_account
+    elif isinstance(pipeline_options, dict):
+      impersonate_service_account = pipeline_options.get(
+          'impersonate_service_account')
     else:
       return credentials
     if impersonate_service_account: