Skip to content

Commit fd27b6a

Browse files
author
Ian Schweer
committed
Remove test value
1 parent 48450b0 commit fd27b6a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/link/pytorch/test_basic.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from pytensor.configdefaults import config
1212
from pytensor.graph.basic import Apply
1313
from pytensor.graph.fg import FunctionGraph
14-
from pytensor.graph.op import Op, get_test_value
14+
from pytensor.graph.op import Op
1515
from pytensor.ifelse import ifelse
1616
from pytensor.raise_op import CheckAndRaise
1717
from pytensor.tensor import alloc, arange, as_tensor, empty, eye
@@ -310,10 +310,9 @@ def test_pytorch_ifelse():
310310

311311
for test_value, cond in [(0.2, 0.5), (0.5, 0.4)]:
312312
a = scalar("a")
313-
a.tag.test_value = np.array(test_value, dtype=config.floatX)
314313
x = ifelse(
315314
a < cond, tuple(np.r_[p1_vals, p2_vals]), tuple(np.r_[p2_vals, p1_vals])
316315
)
317316
x_fg = FunctionGraph([a], x)
318317

319-
compare_pytorch_and_py(x_fg, [get_test_value(i) for i in x_fg.inputs])
318+
compare_pytorch_and_py(x_fg, np.array(test_value, dtype=config.floatX))

0 commit comments

Comments
 (0)