fix bpe files_lim sign

Natooz · Nov 23, 2022 · 2724062 · 2724062
1 parent f92f4aa
commit 2724062
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/miditok/bpe.py b/miditok/bpe.py
@@ -51,7 +51,7 @@ def bpe(self, tokens_path: Union[Path, PurePath, str], vocab_size: int, out_dir:
             out_dir.mkdir(parents=True, exist_ok=True)
             files_paths = list(Path(tokens_path).glob('**/*.json'))
             files_paths_bpe = choices(files_paths, k=files_lim) if \
-                (files_lim is not None and files_lim > len(files_paths)) else files_paths
+                (files_lim is not None and files_lim < len(files_paths)) else files_paths
             samples = []
             samples_paths = []
             original_lengths = []