Removed block_diag test with BlockWise

pymc-devs · Jul 18, 2024 · 76e1471 · 76e1471
1 parent b1b66dc
commit 76e1471
Showing 1 changed file with 0 additions and 16 deletions.
diff --git a/tests/link/pytorch/test_slinalg.py b/tests/link/pytorch/test_slinalg.py
@@ -1,7 +1,6 @@
 import numpy as np
 import pytest
 
-import pytensor.tensor as pt
 from pytensor.configdefaults import config
 from pytensor.graph.fg import FunctionGraph
 from pytensor.tensor import slinalg as pt_slinalg
@@ -128,18 +127,3 @@ def test_pytorch_block_diag():
             np.random.normal(size=(4, 4)).astype(config.floatX),
         ],
     )
-
-
-@pytest.mark.xfail(reason="Blockwise not implemented")
-def test_pytorch_block_diag_blockwise():
-    A = pt.tensor3("A")
-    B = pt.tensor3("B")
-    out = pt_slinalg.block_diag(A, B)
-    out_fg = FunctionGraph([A, B], [out])
-    compare_pytorch_and_py(
-        out_fg,
-        [
-            np.random.normal(size=(5, 5, 5)).astype(config.floatX),
-            np.random.normal(size=(5, 3, 3)).astype(config.floatX),
-        ],
-    )