diff --git a/pyproject.toml b/pyproject.toml index 3fa8dc7..ecadb99 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "dom-tokenizers" -version = "0.0.4" +version = "0.0.5" authors = [{ name = "Gary Benson" }] description = "DOM-aware tokenizers for 🤗 Hugging Face language models" readme = "README.md" @@ -39,6 +39,7 @@ dev = [ "datasets", "flake8", "pytest", + "pytest-cov", ] train = [ "datasets", @@ -53,6 +54,7 @@ requires = ["setuptools>=61.0"] build-backend = "setuptools.build_meta" [tool.pytest.ini_options] +addopts = "--cov=dom_tokenizers" filterwarnings = [ "error", "ignore:`resume_download` is deprecated:FutureWarning", diff --git a/src/dom_tokenizers/internal/transformers.py b/src/dom_tokenizers/internal/transformers.py index 3ddc75c..aabe204 100644 --- a/src/dom_tokenizers/internal/transformers.py +++ b/src/dom_tokenizers/internal/transformers.py @@ -13,6 +13,6 @@ finally: if __orig_val is None: os.environ.pop(__var_name) - else: + else: # pragma: no cover os.environ[__var_name] = __orig_val del __var_name, __orig_val, os