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
         | 

