[BEAM-9034] Update environment_id for ExternalTransform in Python SDK

grtheod · Dec 27, 2019 · ea6ef3c · ea6ef3c
1 parent 614386b
commit ea6ef3c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/transforms/external.py b/sdks/python/apache_beam/transforms/external.py
@@ -419,7 +419,8 @@ def _normalize(coder_proto):
         inputs=self._expanded_transform.inputs,
         outputs={
             tag: pcoll_renames.get(pcoll, pcoll)
-            for tag, pcoll in self._expanded_transform.outputs.items()})
+            for tag, pcoll in self._expanded_transform.outputs.items()},
+        environment_id=self._expanded_transform.environment_id)
 
 
 class JavaJarExpansionService(object):