ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added @needs_torch to test_cuda_oom

Browse source
This commit is contained in:
Roland Tannous 2026-02-11 16:12:37 +00:00
commit 1a6bfe51b6
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
studio/backend/tests/test_utils.py
View file

@ -316,6 +316,7 @@ class TestFormatErrorMessage:
# --- OOM on CUDA ---
@needs_torch
def test_cuda_oom(self):
err = Exception("CUDA out of memory")
with patch("utils.hardware.get_device", return_value=DeviceType.CUDA):