unxfail cases

BasisResearch · Feb 15, 2024 · fd3671c · fd3671c
1 parent 4d7e04b
commit fd3671c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/tests/robust/test_handlers.py b/tests/robust/test_handlers.py
@@ -31,8 +31,11 @@
         lambda m: pyro.infer.autoguide.AutoNormal(pyro.poutine.block(hide=["y"])(m)),
         {"y"},
         1,
-        marks=pytest.mark.xfail(
-            reason="torch.func autograd doesnt work with PyroParam"
+        marks=(
+            [pytest.mark.xfail(reason="torch.func autograd doesnt work with PyroParam")]
+            if tuple(map(int, pyro.__version__.split("+")[0].split(".")[:3]))
+            <= (1, 8, 6)
+            else []
         ),
     ),
 ]

diff --git a/tests/robust/test_ops.py b/tests/robust/test_ops.py
@@ -31,8 +31,11 @@
         lambda m: pyro.infer.autoguide.AutoNormal(pyro.poutine.block(hide=["y"])(m)),
         {"y"},
         1,
-        marks=pytest.mark.xfail(
-            reason="torch.func autograd doesnt work with PyroParam"
+        marks=(
+            [pytest.mark.xfail(reason="torch.func autograd doesnt work with PyroParam")]
+            if tuple(map(int, pyro.__version__.split("+")[0].split(".")[:3]))
+            <= (1, 8, 6)
+            else []
         ),
     ),
 ]