Upload collections_operators.py with huggingface_hub
Browse files- 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
|