Remove test value

pymc-devs · Aug 15, 2024 · ef9277b · ef9277b
1 parent 2766457
commit ef9277b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/link/pytorch/test_basic.py b/tests/link/pytorch/test_basic.py
@@ -11,7 +11,7 @@
 from pytensor.configdefaults import config
 from pytensor.graph.basic import Apply
 from pytensor.graph.fg import FunctionGraph
-from pytensor.graph.op import Op, get_test_value
+from pytensor.graph.op import Op
 from pytensor.ifelse import ifelse
 from pytensor.raise_op import CheckAndRaise
 from pytensor.tensor import alloc, arange, as_tensor, empty, eye
@@ -310,10 +310,9 @@ def test_pytorch_ifelse():
 
     for test_value, cond in [(0.2, 0.5), (0.5, 0.4)]:
         a = scalar("a")
-        a.tag.test_value = np.array(test_value, dtype=config.floatX)
         x = ifelse(
             a < cond, tuple(np.r_[p1_vals, p2_vals]), tuple(np.r_[p2_vals, p1_vals])
         )
         x_fg = FunctionGraph([a], x)
 
-        compare_pytorch_and_py(x_fg, [get_test_value(i) for i in x_fg.inputs])
+        compare_pytorch_and_py(x_fg, np.array(test_value, dtype=config.floatX))