o Zh @s*dZddlmZGdddeZdgZdS)z Processor class for TVLT. )ProcessorMixincsReZdZdZddgZdZdZfddZ     dd d Ze d d Z Z S) TvltProcessora` Constructs a TVLT processor which wraps a TVLT image processor and TVLT feature extractor into a single processor. [`TvltProcessor`] offers all the functionalities of [`TvltImageProcessor`] and [`TvltFeatureExtractor`]. See the docstring of [`~TvltProcessor.__call__`] for more information. Args: image_processor (`TvltImageProcessor`): An instance of [`TvltImageProcessor`]. The image processor is a required input. feature_extractor (`TvltFeatureExtractor`): An instance of [`TvltFeatureExtractor`]. The feature extractor is a required input. image_processorfeature_extractorZTvltImageProcessorZTvltFeatureExtractorcs tj||d||_||_dS)N)rr)super__init__rr)selfrr __class__b/var/www/auris/lib/python3.10/site-packages/transformers/models/deprecated/tvlt/processing_tvlt.pyr(s zTvltProcessor.__init__NFc Os|dur |dur tdd} |dur |j|g|Rd|i|} |dur2|j|g|Rddi|} |durE|j|g|R||d|} i} |durP| | |durY| | | durb| | | S)a Forwards the `images` argument to TvltImageProcessor's [`~TvltImageProcessor.preprocess`] and the `audio` argument to TvltFeatureExtractor's [`~TvltFeatureExtractor.__call__`]. Please refer to the docstring of the above two methods for more information. NzCYou need to specify either an `images` or `audio` input to process. mask_pixelZis_mixedT) sampling_rate mask_audio) ValueErrorrrupdate) rZimagesZaudioZ images_mixedrrr argskwargsZimages_mixed_dictZ images_dictZ audio_dictZ output_dictr r r __call__.s2   zTvltProcessor.__call__cCs"|jj}|jj}tt||S)N)rmodel_input_namesrlistdictfromkeys)rZimage_processor_input_namesZfeature_extractor_input_namesr r r rUszTvltProcessor.model_input_names)NNNNFF) __name__ __module__ __qualname____doc__ attributesZimage_processor_classZfeature_extractor_classrrpropertyr __classcell__r r r r rs   'rN)rZprocessing_utilsrr__all__r r r r s  F