fix learn_bpe in unique_track setting

Natooz · Jul 6, 2023 · 220f384 · 220f384
1 parent de4a6ba
commit 220f384
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/miditok/midi_tokenizer.py b/miditok/midi_tokenizer.py
@@ -1027,7 +1027,7 @@ def learn_bpe(
                 iterator += (
                     [[byte_] for byte_ in bytes_]
                     if not self.unique_track
-                    else [[bytes_]]
+                    else [bytes_]
                 )
 
             # This doesn't seem to work, the trainer pre-processes the sequences, but then no word remains