Avoid using python teststream with prism.

apache · Oct 21, 2024 · 6f5a58e · 6f5a58e
1 parent dc98cfa
commit 6f5a58e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/runners/direct/direct_runner.py b/sdks/python/apache_beam/runners/direct/direct_runner.py
@@ -119,6 +119,9 @@ def accept(self, pipeline):
 
       def visit_transform(self, applied_ptransform):
         transform = applied_ptransform.transform
+        # Python SDK assumes the direct runner TestStream implementation is being used.
+        if isinstance(transform, TestStream):
+          self.supported_by_prism_runner = False
         if isinstance(transform, beam.ParDo):
           dofn = transform.dofn
           # It's uncertain if the Prism Runner supports execution of CombineFns