WIP

datamol-io · Oct 28, 2023 · e0cbede · e0cbede
1 parent 126c655
commit e0cbede
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -77,7 +77,7 @@ include = '\.pyi?$'
 
 [tool.pytest.ini_options]
 minversion = "6.0"
-addopts = "--verbose --color yes --disable-warnings"
+addopts = "--verbose --color yes"
 testpaths = ["tests"]
 
 [tool.ruff.pycodestyle]

diff --git a/safe/trainer/collator.py b/safe/trainer/collator.py
@@ -12,7 +12,7 @@
 from transformers.data.data_collator import _torch_collate_batch
 from safe.tokenizer import SAFETokenizer
 
-from tokenizer import Tokenizer
+from tokenizers import Tokenizer
 
 
 class SAFECollator: