commit 8ba16fad4cfa3f829922c6b490f8a96f06f4ba7a
author Robert Bradshaw <robertwb@gmail.com> Tue May 11 12:04:08 2021 -0700
committer Robert Bradshaw <robertwb@gmail.com> Tue May 11 12:04:08 2021 -0700
tree 0a7981276c2eee75a924100a293d1e00abc31866
parent ae22aed92143de834cd1b066379cb573a373646f

[BEAM-12326] Preserve resource hints across transform substitution.

sdks/python/apache_beam/pipeline.py

diff --git a/sdks/python/apache_beam/pipeline.py b/sdks/python/apache_beam/pipeline.py
index e296a30..35d38a7 100644
--- a/sdks/python/apache_beam/pipeline.py
+++ b/sdks/python/apache_beam/pipeline.py
@@ -299,6 +299,9 @@
               original_transform_node.full_label,
               original_transform_node.inputs)
 
+          replacement_transform_node.resource_hints = (
+              original_transform_node.resource_hints)
+
           # Transform execution could depend on order in which nodes are
           # considered. Hence we insert the replacement transform node to same
           # index as the original transform node. Note that this operation