Avoid re-encoding row types. (#12491)

There is no benefit in decoding these only to recode them to pass them back to the SDK.
apache · Sep 17, 2020 · f4c2734 · f4c2734
2 parents 51cb377 + d783c7a
commit f4c2734
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/runners/portability/fn_api_runner/translations.py b/sdks/python/apache_beam/runners/portability/fn_api_runner/translations.py
@@ -351,8 +351,10 @@ def wrapper(self, *args):
 class TransformContext(object):
 
   _KNOWN_CODER_URNS = set(
-      value.urn for key,
-      value in common_urns.coders.__dict__.items() if not key.startswith('_'))
+      value.urn for (key, value) in common_urns.coders.__dict__.items()
+      if not key.startswith('_')
+      # Length prefix Rows rather than re-coding them.
+  ) - set([common_urns.coders.ROW.urn])
 
   def __init__(
       self,