Skip to content

Commit 76e1471

Browse files
committed
Removed block_diag test with BlockWise
1 parent b1b66dc commit 76e1471

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tests/link/pytorch/test_slinalg.py

-16
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import numpy as np
22
import pytest
33

4-
import pytensor.tensor as pt
54
from pytensor.configdefaults import config
65
from pytensor.graph.fg import FunctionGraph
76
from pytensor.tensor import slinalg as pt_slinalg
@@ -128,18 +127,3 @@ def test_pytorch_block_diag():
128127
np.random.normal(size=(4, 4)).astype(config.floatX),
129128
],
130129
)
131-
132-
133-
@pytest.mark.xfail(reason="Blockwise not implemented")
134-
def test_pytorch_block_diag_blockwise():
135-
A = pt.tensor3("A")
136-
B = pt.tensor3("B")
137-
out = pt_slinalg.block_diag(A, B)
138-
out_fg = FunctionGraph([A, B], [out])
139-
compare_pytorch_and_py(
140-
out_fg,
141-
[
142-
np.random.normal(size=(5, 5, 5)).astype(config.floatX),
143-
np.random.normal(size=(5, 3, 3)).astype(config.floatX),
144-
],
145-
)

0 commit comments

Comments
 (0)