Elron commited on
Commit
c5f8a6a
1 Parent(s): 5f04977

Upload collections_operators.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. collections_operators.py +9 -0
collections_operators.py CHANGED
@@ -32,6 +32,15 @@ class Wrap(FieldOperator):
32
  }
33
 
34
 
 
 
 
 
 
 
 
 
 
35
  class Slice(FieldOperator):
36
  start: Optional[int] = None
37
  stop: Optional[int] = None
 
32
  }
33
 
34
 
35
+ class Chunk(FieldOperator):
36
+ size: int
37
+
38
+ def process_value(self, collection: Any) -> Any:
39
+ return [
40
+ collection[i : i + self.size] for i in range(0, len(collection), self.size)
41
+ ]
42
+
43
+
44
  class Slice(FieldOperator):
45
  start: Optional[int] = None
46
  stop: Optional[int] = None