gentoo-mirror/gentoo
1
0
Fork 0
mirror of https://github.com/gentoo-mirror/gentoo.git synced 2026-01-06 02:17:34 -08:00
Code Issues Projects Releases Wiki Activity

sci-ml/accelerate: add test deps

Browse Source 
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
This commit is contained in:
Alfredo Tupone 2025-05-21 11:43:51 +02:00
parent 2d06936214
commit 09cda66272
No known key found for this signature in database
GPG Key ID: FBE5925C5B02CE58
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
sci-ml/accelerate/accelerate-1.6.0-r4.ebuild → sci-ml/accelerate/accelerate-1.6.0-r5.ebuild
View File

@ -35,25 +35,19 @@ BDEPEND="test? (
sci-ml/clearml[${PYTHON_USEDEP}]
')
sci-ml/caffe2[gloo]
sci-ml/evaluate[${PYTHON_SINGLE_USEDEP}]
sci-ml/torchdata[${PYTHON_SINGLE_USEDEP}]
)"
distutils_enable_tests pytest
python_test() {
local EPYTEST_IGNORE=(
tests/test_examples.py
tests/deepspeed
tests/test_big_modeling.py
tests/fsdp
tests/tp
tests/test_cli.py
)
local EPYTEST_DESELECT=(
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_buffer_check
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_buffer_check_and_multi_devices
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_fallback_allocation_and_buffers
tests/test_utils.py::UtilsTester::test_patch_environment_key_exists
tests/test_examples.py::FeatureExamplesTests
)
epytest tests
}