SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

rm require_version_examples (#12088)

Browse Source
This commit is contained in:
Stas Bekman
2021-06-09 11:02:52 -07:00
committed by GitHub
parent d1500d9151
commit 61e191987d
6 changed files with 9 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/legacy/pytorch-lightning/lightning_base.py
View File

@@ -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,

4
examples/research_projects/rag-end2end-retriever/lightning_base.py
View File

@@ -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,

4
examples/research_projects/rag/lightning_base.py
View File

@@ -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,

4
examples/research_projects/seq2seq-distillation/lightning_base.py
View File

@@ -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,

6
src/transformers/utils/versions.py
View File

@@ -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
tests/test_versions_utils.py
View File

@@ -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):
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