コード例 #1
0
ファイル: dataset.py プロジェクト: DmitriGekhtman/ray
 def transform(block: Block) -> Block:
     block = BlockAccessor.for_block(block)
     builder = block.builder()
     for row in block.iter_rows():
         if fn(row):
             builder.add(row)
     return builder.build()
コード例 #2
0
ファイル: dataset.py プロジェクト: DmitriGekhtman/ray
 def transform(block: Block) -> Block:
     block = BlockAccessor.for_block(block)
     builder = DelegatingArrowBlockBuilder()
     for row in block.iter_rows():
         for r2 in fn(row):
             builder.add(r2)
     return builder.build()
コード例 #3
0
ファイル: dataset.py プロジェクト: DmitriGekhtman/ray
 def agg(block: Block) -> int:
     block = BlockAccessor.for_block(block)
     return sum(block.iter_rows())