[fix] cannot import name 'Pop2PianoFeatureExtractor' from 'transforme…

…rs' (#35604) * update pop2piano __init__ * add lib check * update fix * revert
huggingface · Jan 20, 2025 · 3998fa8 · 3998fa8
1 parent b80e334
commit 3998fa8
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/transformers/models/pop2piano/feature_extraction_pop2piano.py b/src/transformers/models/pop2piano/feature_extraction_pop2piano.py
@@ -448,3 +448,6 @@ def __call__(
         )
 
         return output
+
+
+__all__ = ["Pop2PianoFeatureExtractor"]
diff --git a/src/transformers/models/pop2piano/processing_pop2piano.py b/src/transformers/models/pop2piano/processing_pop2piano.py
@@ -137,3 +137,6 @@ def save_pretrained(self, save_directory, **kwargs):
     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
@@ -714,3 +714,6 @@ def batch_decode(
             return BatchEncoding({"notes": notes_list, "pretty_midi_objects": pretty_midi_objects_list})
 
         return BatchEncoding({"notes": notes_list})
+
+
+__all__ = ["Pop2PianoTokenizer"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -448,3 +448,6 @@ def __call__(
		)

		return output


		__all__ = ["Pop2PianoFeatureExtractor"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -714,3 +714,6 @@ def batch_decode(
		return BatchEncoding({"notes": notes_list, "pretty_midi_objects": pretty_midi_objects_list})

		return BatchEncoding({"notes": notes_list})


		__all__ = ["Pop2PianoTokenizer"]