[versions] handle version requirement ranges (#11110)

* handle version requirement ranges

* add mixed requirement test

* cleanup
This commit is contained in:
Stas Bekman
2021-04-07 09:09:38 -07:00
committed by GitHub
parent 7442801df5
commit 1c15128312
2 changed files with 35 additions and 21 deletions

View File

@@ -14,8 +14,6 @@
import sys
import numpy
from transformers.testing_utils import TestCasePlus
from transformers.utils.versions import (
importlib_metadata,
@@ -25,7 +23,7 @@ from transformers.utils.versions import (
)
numpy_ver = numpy.__version__
numpy_ver = importlib_metadata.version("numpy")
python_ver = ".".join([str(x) for x in sys.version_info[:3]])
@@ -54,6 +52,9 @@ class DependencyVersionCheckTest(TestCasePlus):
# gt
require_version_core("numpy>1.0.0")
# mix
require_version_core("numpy>1.0.0,<1000")
# requirement w/o version
require_version_core("numpy")