Method __call__

detrsmpl/data/datasets/pipelines/formatting.py:40–43

(self, results)

        self.keys = keys

    def __call__(self, results):
        for key in self.keys:
            results[key] = to_tensor(results[key])
        return results

    def __repr__(self):
        return self.__class__.__name__ + f'(keys={self.keys})'