use as_tensor_variable instead

williambdean · williambdean · commit 1ab1dc3c2b6e · 2025-07-27T19:47:37.000-04:00
diff --git a/tests/tensor/rewriting/test_math.py b/tests/tensor/rewriting/test_math.py
@@ -1997,7 +1997,7 @@ def test_log_sqrt() -> None:
 
     assert equal_computations(
         [out],
-        [mul(np.array([[0.5]]), log(x))],
+        [mul(pt.as_tensor_variable([[0.5]]), log(x))],
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -1997,7 +1997,7 @@ def test_log_sqrt() -> None:`
`1997`	`1997`
`1998`	`1998`	`assert equal_computations(`
`1999`	`1999`	`[out],`
`2000`		`- [mul(np.array([[0.5]]), log(x))],`
	`2000`	`+ [mul(pt.as_tensor_variable([[0.5]]), log(x))],`
`2001`	`2001`	`)`
`2002`	`2002`
`2003`	`2003`