fix bug

joao-alex-cunha · rwightman · commit a38c2b447595 · 2023-11-16T20:27:59.000-08:00
diff --git a/tests/test_optim.py b/tests/test_optim.py
@@ -106,7 +106,7 @@ def fn_base(optimizer, weight, bias):
     # to a different type and move to a different device.
     if torch_device == 'cpu':
         return
-    elif torch_device == 'cuda' and not torch.cuda.is_available:
+    elif torch_device == 'cuda' and not torch.cuda.is_available():
         return
 
     with torch.no_grad():
