| 48 | |
| 49 | @PIPELINES.register_module() |
| 50 | class Transpose(object): |
| 51 | |
| 52 | def __init__(self, keys, order): |
| 53 | self.keys = keys |
| 54 | self.order = order |
| 55 | |
| 56 | def __call__(self, results): |
| 57 | for key in self.keys: |
| 58 | results[key] = results[key].transpose(self.order) |
| 59 | return results |
| 60 | |
| 61 | def __repr__(self): |
| 62 | return self.__class__.__name__ + \ |
| 63 | f'(keys={self.keys}, order={self.order})' |
| 64 | |
| 65 | |
| 66 | @PIPELINES.register_module() |
nothing calls this directly
no outgoing calls
no test coverage detected