[ 
https://issues.apache.org/jira/browse/BEAM-14219?focusedWorklogId=759606&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-759606
 ]

ASF GitHub Bot logged work on BEAM-14219:
-----------------------------------------

                Author: ASF GitHub Bot
            Created on: 20/Apr/22 21:44
            Start Date: 20/Apr/22 21:44
    Worklog Time Spent: 10m 
      Work Description: y1chi commented on code in PR #17323:
URL: https://github.com/apache/beam/pull/17323#discussion_r854578401


##########
sdks/python/apache_beam/runners/dataflow/dataflow_runner.py:
##########
@@ -393,6 +393,72 @@ def _adjust_pipeline_for_dataflow_v2(self, pipeline):
   def _check_for_unsupported_features_on_non_portable_worker(self, pipeline):
     pipeline.visit(self.combinefn_visitor())
 
+  def _enable_sdk_prebuild_if_applicable(self, options):
+    setup_options = options.view_as(SetupOptions)
+    # SDK container image pre-build explicitly enabled or disabled.
+    if (setup_options.prebuild_sdk_container_engine or
+        setup_options.disable_sdk_container_prebuild):
+      return
+    from apache_beam.runners.dataflow.internal import apiclient
+    # prebuild_sdk_container_engine only works with fnapi jobs.
+    if (not apiclient._use_fnapi(options) and

Review Comment:
   On a side note, I guess the prebuilding doesn't really involve beam_fn_api, 
so I'm wondering if we can backport the boot.go change to non-fnapi python 
worker containers, I think the prebuilding has value for jobs on legacy python 
batch worker as well, WDYT?





Issue Time Tracking
-------------------

    Worklog Id:     (was: 759606)
    Time Spent: 6h  (was: 5h 50m)

> Dataflow runner prebuilds SDK container with cloud build by default
> -------------------------------------------------------------------
>
>                 Key: BEAM-14219
>                 URL: https://issues.apache.org/jira/browse/BEAM-14219
>             Project: Beam
>          Issue Type: Improvement
>          Components: runner-dataflow
>            Reporter: Yichi Zhang
>            Assignee: Yichi Zhang
>            Priority: P2
>          Time Spent: 6h
>  Remaining Estimate: 0h
>




--
This message was sent by Atlassian Jira
(v8.20.7#820007)

Reply via email to