423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-17 04:10:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] lint (#9593)

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Yaowei Zheng
2025-12-09 18:00:35 +08:00
committed by GitHub
parent 1bbb461f76
commit 5d56817e2b
23 changed files with 51 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/conftest.py
View File

@@ -40,9 +40,7 @@ def pytest_configure(config):
config.addinivalue_line(
"markers", "require_device: test requires specific device, e.g., @pytest.mark.require_device('cuda')"
)
config.addinivalue_line(
"markers", "runs_on: test requires specific device, e.g., @pytest.mark.runs_on(['cpu'])"
)
config.addinivalue_line("markers", "runs_on: test requires specific device, e.g., @pytest.mark.runs_on(['cpu'])")
def _handle_runs_on(items):
@@ -64,14 +62,12 @@ def _handle_runs_on(items):
if isinstance(runs_on_devices, str):
runs_on_devices = [runs_on_devices]
if CURRENT_DEVICE not in runs_on_devices:
item.add_marker(
pytest.mark.skip(
reason=f"test requires one of {runs_on_devices} (current: {CURRENT_DEVICE})"
)
pytest.mark.skip(reason=f"test requires one of {runs_on_devices} (current: {CURRENT_DEVICE})")
)
def _handle_slow_tests(items):
"""Skip slow tests unless RUN_SLOW environment variable is set.