[api] Fixes unitest (#3210)

The test was failing on the machine installed cuda but no GPU
deepjavalibrary · May 28, 2024 · d241860 · d241860
1 parent 97c6a0d
commit d241860
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/api/src/test/java/ai/djl/util/cuda/CudaUtilsTest.java b/api/src/test/java/ai/djl/util/cuda/CudaUtilsTest.java
@@ -28,7 +28,11 @@ public class CudaUtilsTest {
     @Test
     public void testCudaUtils() {
         if (!CudaUtils.hasCuda()) {
-            Assert.assertThrows(CudaUtils::getCudaVersionString);
+            try {
+                CudaUtils.getCudaVersionString();
+            } catch (IllegalStateException | IllegalArgumentException ignore) {
+                // ignore exception
+            }
             Assert.assertThrows(() -> CudaUtils.getComputeCapability(0));
             Assert.assertThrows(() -> CudaUtils.getGpuMemory(Device.gpu()));
             return;