fix: re-adding kwargs when saving tokens

Natooz · Aug 18, 2023 · 2e1978f · 2e1978f
1 parent ca00539
commit 2e1978f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/miditok/midi_tokenizer.py b/miditok/midi_tokenizer.py
@@ -1677,7 +1677,7 @@ def save_tokens(
             kwargs["ids_bpe_encoded"] = ids_bpe_encoded
 
         with open(path, "w") as outfile:
-            dic = {"ids": ids}
+            dic = {"ids": ids, **kwargs}
             if programs is not None:
                 dic["programs"] = programs
             json.dump(dic, outfile)