Do not test pickling native sink objects

apache · Dec 13, 2016 · e36ee8d · e36ee8d
1 parent f086afe
commit e36ee8d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdks/python/apache_beam/runners/direct/transform_evaluator.py b/sdks/python/apache_beam/runners/direct/transform_evaluator.py
@@ -505,8 +505,7 @@ def __init__(self, evaluation_context, applied_ptransform,
         side_inputs)
 
     assert applied_ptransform.transform.sink
-    # TODO(aaltay): Consider storing the serialized form as an optimization.
-    self._sink = pickler.loads(pickler.dumps(applied_ptransform.transform.sink))
+    self._sink = applied_ptransform.transform.sink
 
   @property
   def _is_final_bundle(self):