riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

add test marker to run all tests with @require_bitsandbytes (#28278)

Browse Source
This commit is contained in:
Titus 2024-02-15 21:53:09 -03:00 committed by GitHub
parent f3aa7db439
commit 1e402b957d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -32,4 +32,5 @@ doctest_optionflags="NUMBER NORMALIZE_WHITESPACE ELLIPSIS"
doctest_glob="**/*.md" doctest_glob="**/*.md"
markers = [ markers = [
"flash_attn_test: marks tests related to flash attention (deselect with '-m \"not flash_attn_test\"')", "flash_attn_test: marks tests related to flash attention (deselect with '-m \"not flash_attn_test\"')",
"bitsandbytes: select (or deselect with `not`) bitsandbytes integration tests",
] ]

12
src/transformers/testing_utils.py
View File

@ -966,9 +966,17 @@ def require_aqlm(test_case):
def require_bitsandbytes(test_case): def require_bitsandbytes(test_case):
""" """
Decorator for bits and bytes (bnb) dependency Decorator marking a test that requires the bitsandbytes library. Will be skipped when the library or its hard dependency torch is not installed.
""" """
return unittest.skipUnless(is_bitsandbytes_available(), "test requires bnb")(test_case) if is_bitsandbytes_available() and is_torch_available():
try:
import pytest
return pytest.mark.bitsandbytes(test_case)
except ImportError:
return test_case
else:
return unittest.skip("test requires bitsandbytes and torch")(test_case)
def require_optimum(test_case): def require_optimum(test_case):