diff --git a/studio/backend/tests/test_utils.py b/studio/backend/tests/test_utils.py index 60a9969edc..afb4bab65a 100644 --- a/studio/backend/tests/test_utils.py +++ b/studio/backend/tests/test_utils.py @@ -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):