riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 9 additions and 28 deletions
Show Stats 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_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"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_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"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_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"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_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"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"""
hint = "Try: pip install transformers -U or pip install -e '.[dev]' if you're working with git master"
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
from transformers.testing_utils import TestCasePlus
from transformers.utils.versions import (
importlib_metadata,
require_version,
require_version_core,
require_version_examples,
)
from transformers.utils.versions import importlib_metadata, require_version, require_version_core
numpy_ver = importlib_metadata.version("numpy")
@ -88,14 +83,6 @@ class DependencyVersionCheckTest(TestCasePlus):
except ValueError as 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):
# matching requirement