dev(hansbug): fix bug in unittest

opendilab · Oct 20, 2023 · 4f9a9f1 · 4f9a9f1
1 parent 1a23d12
commit 4f9a9f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/tree/integration/test_torch.py b/test/tree/integration/test_torch.py
@@ -162,7 +162,7 @@ def forward(self, x):
             'd': torch.Size([2, 5, 3]),
         })
 
-    @skipUnless(vpip('torch') and OS.linux and vpython < '3.11', 'torch required')
+    @skipUnless(torch is not None and vpip('torch') and OS.linux and vpython < '3.11', 'torch required')
     def test_moduledict(self):
         with torch.no_grad():
             md = torch.nn.ModuleDict({