Disable a test that uses torch.compile in python 3.12

torchmd · Jun 10, 2024 · dbc1681 · dbc1681
1 parent 8e6ea00
commit dbc1681
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_neighbors.py b/tests/test_neighbors.py
@@ -669,6 +669,10 @@ def test_per_batch_box(device, strategy, n_batches, use_forward):
 @pytest.mark.parametrize("loop", [True, False])
 @pytest.mark.parametrize("include_transpose", [True, False])
 def test_torch_compile(device, dtype, loop, include_transpose):
+    import sys
+
+    if sys.version_info >= (3, 12):
+        pytest.skip("Not available in this version")
     if torch.__version__ < "2.0.0":
         pytest.skip("Not available in this version")
     if device == "cuda" and not torch.cuda.is_available():