From 1a6bfe51b620da2c491756441b42d594fcc841ed Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Wed, 11 Feb 2026 16:12:37 +0000 Subject: [PATCH] added @needs_torch to test_cuda_oom --- studio/backend/tests/test_utils.py | 1 + 1 file changed, 1 insertion(+) 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):