tokenizing dataset pbar now displays out_dir name

Natooz · Dec 19, 2022 · b3b0cc7 · b3b0cc7
1 parent 4b82548
commit b3b0cc7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/miditok/midi_tokenizer_base.py b/miditok/midi_tokenizer_base.py
@@ -516,8 +516,8 @@ def tokenize_midi_dataset(self, midi_paths: Union[List[str], List[Path], List[Pu
         out_dir.mkdir(parents=True, exist_ok=True)
         self.save_params(out_dir / 'config.txt')  # Saves the parameters with which the MIDIs are converted
 
-        for midi_path in tqdm(midi_paths, desc='Converting MIDIs to tokens') if logging else enumerate(midi_paths):
-            # Some MIDIs can contains errors that are raised by Mido, if so the loop continues
+        for midi_path in tqdm(midi_paths, desc=f'Tokenizing MIDIs ({out_dir.name})') if logging else midi_paths:
+            # Some MIDIs can contain errors that are raised by Mido, if so the loop continues
             try:
                 midi = MidiFile(PurePath(midi_path))
             except FileNotFoundError: