Unverified Commit 61e19198 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

rm require_version_examples (#12088)

parent d1500d91
...@@ -28,12 +28,12 @@ from transformers.optimization import ( ...@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup, get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup, get_polynomial_decay_schedule_with_warmup,
) )
from transformers.utils.versions import require_version_examples from transformers.utils.versions import require_version
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4") require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = { MODEL_MODES = {
"base": AutoModel, "base": AutoModel,
......
...@@ -29,12 +29,12 @@ from transformers.optimization import ( ...@@ -29,12 +29,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup, get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup, get_polynomial_decay_schedule_with_warmup,
) )
from transformers.utils.versions import require_version_examples from transformers.utils.versions import require_version
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4") require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = { MODEL_MODES = {
"base": AutoModel, "base": AutoModel,
......
...@@ -28,12 +28,12 @@ from transformers.optimization import ( ...@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup, get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup, get_polynomial_decay_schedule_with_warmup,
) )
from transformers.utils.versions import require_version_examples from transformers.utils.versions import require_version
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4") require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = { MODEL_MODES = {
"base": AutoModel, "base": AutoModel,
......
...@@ -28,12 +28,12 @@ from transformers.optimization import ( ...@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup, get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup, get_polynomial_decay_schedule_with_warmup,
) )
from transformers.utils.versions import require_version_examples from transformers.utils.versions import require_version
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4") require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = { MODEL_MODES = {
"base": AutoModel, "base": AutoModel,
......
...@@ -118,9 +118,3 @@ def require_version_core(requirement): ...@@ -118,9 +118,3 @@ def require_version_core(requirement):
"""require_version wrapper which emits a core-specific hint on failure""" """require_version wrapper which emits a core-specific hint on failure"""
hint = "Try: pip install transformers -U or pip install -e '.[dev]' if you're working with git master" hint = "Try: pip install transformers -U or pip install -e '.[dev]' if you're working with git master"
return require_version(requirement, hint) return require_version(requirement, hint)
def require_version_examples(requirement):
"""require_version wrapper which emits examples-specific hint on failure"""
hint = "Try: pip install -r examples/requirements.txt"
return require_version(requirement, hint)
...@@ -15,12 +15,7 @@ ...@@ -15,12 +15,7 @@
import sys import sys
from transformers.testing_utils import TestCasePlus from transformers.testing_utils import TestCasePlus
from transformers.utils.versions import ( from transformers.utils.versions import importlib_metadata, require_version, require_version_core
importlib_metadata,
require_version,
require_version_core,
require_version_examples,
)
numpy_ver = importlib_metadata.version("numpy") numpy_ver = importlib_metadata.version("numpy")
...@@ -88,14 +83,6 @@ class DependencyVersionCheckTest(TestCasePlus): ...@@ -88,14 +83,6 @@ class DependencyVersionCheckTest(TestCasePlus):
except ValueError as e: except ValueError as e:
self.assertIn("need one of ", str(e)) self.assertIn("need one of ", str(e))
def test_examples(self):
# the main functionality is tested in `test_core`, this is just the hint check
try:
require_version_examples("numpy>1000.4.5")
except ImportError as e:
self.assertIn("is required", str(e))
self.assertIn("pip install -r examples/requirements.txt", str(e))
def test_python(self): def test_python(self):
# matching requirement # matching requirement
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment