| # TODO(karita): add this to all the transform impl. | |
| class TransformInterface: | |
| """Transform Interface""" | |
| def __call__(self, x): | |
| raise NotImplementedError("__call__ method is not implemented") | |
| def add_arguments(cls, parser): | |
| return parser | |
| def __repr__(self): | |
| return self.__class__.__name__ + "()" | |
| class Identity(TransformInterface): | |
| """Identity Function""" | |
| def __call__(self, x): | |
| return x | |