mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-11 14:49:43 +00:00
fix(ci): test only multi-gpu tests in multi-gpu runner (#3092)
This commit is contained in:
@@ -206,5 +206,4 @@ jobs:
|
|||||||
python -c "import torch; print(f'PyTorch CUDA available: {torch.cuda.is_available()}'); print(f'Number of GPUs: {torch.cuda.device_count()}')"
|
python -c "import torch; print(f'PyTorch CUDA available: {torch.cuda.is_available()}'); print(f'Number of GPUs: {torch.cuda.device_count()}')"
|
||||||
|
|
||||||
- name: Run multi-GPU training tests
|
- name: Run multi-GPU training tests
|
||||||
# TODO(Steven): Investigate why motors tests are failing in multi-GPU setup
|
run: pytest -vv tests/training/
|
||||||
run: pytest tests -vv --maxfail=10 --ignore=tests/motors/
|
|
||||||
|
|||||||
Reference in New Issue
Block a user