fix cpword token type graph (adding ignore key)

Natooz · May 18, 2023 · 9970ec4 · 9970ec4
1 parent 4cf8cb8
commit 9970ec4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/miditok/tokenizations/cp_word.py b/miditok/tokenizations/cp_word.py
@@ -497,6 +497,10 @@ def _create_token_types_graph(self) -> Dict[str, List[str]]:
             dic["Rest"] = ["Rest", "Position", "Bar"]
             dic["Pitch"] += ["Rest"]
 
+        for key in dic:
+            dic[key].append("Ignore")
+        dic["Ignore"] = list(dic.keys())
+
         return dic
 
     @_in_as_seq()