diff --git a/src/transformers/models/pop2piano/feature_extraction_pop2piano.py b/src/transformers/models/pop2piano/feature_extraction_pop2piano.py index 738b93235..89fb04907 100644 --- a/src/transformers/models/pop2piano/feature_extraction_pop2piano.py +++ b/src/transformers/models/pop2piano/feature_extraction_pop2piano.py @@ -448,3 +448,6 @@ class Pop2PianoFeatureExtractor(SequenceFeatureExtractor): ) return output + + +__all__ = ["Pop2PianoFeatureExtractor"] diff --git a/src/transformers/models/pop2piano/processing_pop2piano.py b/src/transformers/models/pop2piano/processing_pop2piano.py index 280e5dc79..11afcab31 100644 --- a/src/transformers/models/pop2piano/processing_pop2piano.py +++ b/src/transformers/models/pop2piano/processing_pop2piano.py @@ -137,3 +137,6 @@ class Pop2PianoProcessor(ProcessorMixin): def from_pretrained(cls, pretrained_model_name_or_path, **kwargs): args = cls._get_arguments_from_pretrained(pretrained_model_name_or_path, **kwargs) return cls(*args) + + +__all__ = ["Pop2PianoProcessor"] diff --git a/src/transformers/models/pop2piano/tokenization_pop2piano.py b/src/transformers/models/pop2piano/tokenization_pop2piano.py index 5ad0996c1..18adb2e96 100644 --- a/src/transformers/models/pop2piano/tokenization_pop2piano.py +++ b/src/transformers/models/pop2piano/tokenization_pop2piano.py @@ -714,3 +714,6 @@ class Pop2PianoTokenizer(PreTrainedTokenizer): return BatchEncoding({"notes": notes_list, "pretty_midi_objects": pretty_midi_objects_list}) return BatchEncoding({"notes": notes_list}) + + +__all__ = ["Pop2PianoTokenizer"]