| 7 | |
| 8 | |
| 9 | class ProcessorWrapper: |
| 10 | def __init__(self, processor): |
| 11 | self.processor = processor |
| 12 | |
| 13 | def __call__(self, tensor): |
| 14 | return self.processor(tensor, return_tensors="pt")["pixel_values"].squeeze(0) |
| 15 | |
| 16 | |
| 17 | def possible_override_args(override_args, *args): |
nothing calls this directly
no outgoing calls
no test coverage detected