Merge pull request #14899 from y1chi/prebuild_timeout

Minor fix to prebuilding sdk workflow timeout setting
apache · Jun 1, 2021 · b39ff90 · b39ff90
2 parents d40abe7 + 50e6249
commit b39ff90
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdks/python/apache_beam/runners/portability/sdk_container_builder.py b/sdks/python/apache_beam/runners/portability/sdk_container_builder.py
@@ -36,7 +36,6 @@
 import uuid
 from typing import Type
 
-from google.protobuf.duration_pb2 import Duration
 from google.protobuf.json_format import MessageToJson
 
 from apache_beam import version as beam_version
@@ -259,7 +258,7 @@ def _invoke_docker_build_and_push(self, container_image_name):
     source.storageSource.object = gcs_object
     build.source = source
     # TODO(zyichi): make timeout configurable
-    build.timeout = Duration().FromSeconds(seconds=7200)
+    build.timeout = '7200s'
 
     now = time.time()
     # operation = client.create_build(project_id=project_id, build=build)