dev(narugo): torch.compile not support py3.11 yet

opendilab · Aug 10, 2023 · e0c78e2 · e0c78e2
1 parent 88705f3
commit e0c78e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/tree/integration/test_torch.py b/test/tree/integration/test_torch.py
@@ -1,7 +1,7 @@
 from unittest import skipUnless
 
 import pytest
-from hbutils.testing import vpip, OS
+from hbutils.testing import vpip, OS, vpython
 
 from treevalue import FastTreeValue, register_for_torch
 
@@ -62,7 +62,7 @@ class MyTreeValueX(FastTreeValue):
         with pytest.warns(UserWarning):
             register_for_torch(MyTreeValueX)
 
-    @skipUnless(vpip('torch') >= '2.0.0' and OS.linux, 'Torch 2 on linux platform required')
+    @skipUnless(vpip('torch') >= '2.0.0' and OS.linux and vpython < '3.11', 'Torch 2 on linux platform required')
     def test_torch_compile(self):
         @torch.compile
         def foo(x, y, t):