test: Fix test_tensor_used_multiple_times warning (#403)

ValerianRey · web-flow · commit 3ef5c3fe10fd · 2025-08-11T11:54:20.000+02:00
* Before this, we had `UserWarning: Converting a tensor with requires_grad=True to a scalar may lead to unexpected behavior`.
diff --git a/tests/unit/autojac/test_backward.py b/tests/unit/autojac/test_backward.py
@@ -233,8 +233,8 @@ def test_tensor_used_multiple_times(chunk_size: int | None):
 
     expected_jacobian = tensor_(
         [
-            [2.0 * 3.0 * a**2],
-            [2.0 * 4.0 * a**3],
+            [2.0 * 3.0 * (a**2).item()],
+            [2.0 * 4.0 * (a**3).item()],
         ],
     )
 

Original file line number	Diff line number	Diff line change
`@@ -233,8 +233,8 @@ def test_tensor_used_multiple_times(chunk_size: int \| None):`
`233`	`233`
`234`	`234`	`expected_jacobian = tensor_(`
`235`	`235`	`[`
`236`		`- [2.0 * 3.0 * a**2],`
`237`		`- [2.0 * 4.0 * a**3],`
	`236`	`+ [2.0 * 3.0 * (a**2).item()],`
	`237`	`+ [2.0 * 4.0 * (a**3).item()],`
`238`	`238`	`],`
`239`	`239`	`)`
`240`	`240`