| """Sonata processor: ``SonataFeatureExtractor`` (the package's inference transform) as its only part.""" | |
| from transformers.processing_utils import ProcessorMixin | |
| class SonataProcessor(ProcessorMixin): | |
| attributes = ["feature_extractor"] | |
| def __init__(self, feature_extractor=None, **kwargs): | |
| super().__init__(feature_extractor, **kwargs) | |
| def __call__(self, point=None, **kwargs): | |
| return self.feature_extractor(point, **kwargs) | |
| __all__ = ["SonataProcessor"] | |