[versions] handle version requirement ranges (#11110)
* handle version requirement ranges * add mixed requirement test * cleanup
This commit is contained in:
@@ -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")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user