コード例 #1
0
 def get_replacement_transform(self, ptransform):
     assert isinstance(ptransform, ParDo)
     do_fn = ptransform.fn
     signature = DoFnSignature(do_fn)
     if signature.is_splittable_dofn():
         return SplittableParDo(ptransform)
     else:
         return ptransform
コード例 #2
0
ファイル: sdf_common.py プロジェクト: aaltay/incubator-beam 
 def get_replacement_transform(self, ptransform):
   assert isinstance(ptransform, ParDo)
   do_fn = ptransform.fn
   signature = DoFnSignature(do_fn)
   if signature.is_splittable_dofn():
     return SplittableParDo(ptransform)
   else:
     return ptransform
コード例 #3
0
 def matches(self, applied_ptransform):
     assert isinstance(applied_ptransform, AppliedPTransform)
     transform = applied_ptransform.transform
     if isinstance(transform, ParDo):
         signature = DoFnSignature(transform.fn)
         return signature.is_splittable_dofn()
コード例 #4
0
ファイル: sdf_common.py プロジェクト: aaltay/incubator-beam 
 def _matcher(applied_ptransform):
   assert isinstance(applied_ptransform, AppliedPTransform)
   transform = applied_ptransform.transform
   if isinstance(transform, ParDo):
     signature = DoFnSignature(transform.fn)
     return signature.is_splittable_dofn()