token types graph Remi / CP Word fix

Natooz · Nov 18, 2021 · 9457178 · 9457178
1 parent af90e72
commit 9457178
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/miditok/cp_word.py b/miditok/cp_word.py
@@ -350,7 +350,6 @@ def _create_token_types_graph(self) -> Dict[str, List[str]]:
         if self.additional_tokens['Rest']:
             dic['Rest'] = ['Rest', 'Position', 'Bar']
             dic['Pitch'] += ['Rest']
-            dic['Bar'] += ['Rest']
 
         return dic
 

diff --git a/miditok/remi.py b/miditok/remi.py
@@ -267,7 +267,6 @@ def _create_token_types_graph(self) -> Dict[str, List[str]]:
         if self.additional_tokens['Rest']:
             dic['Rest'] = ['Rest', 'Position', 'Bar']
             dic['Duration'] += ['Rest']
-            dic['Bar'] += ['Rest']
 
         return dic