We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f0fe529 commit cff6d8eCopy full SHA for cff6d8e
1 file changed
tests/unit/autojac/test_backward.py
@@ -236,9 +236,9 @@ def test_tensor_used_multiple_times(chunk_size: int | None):
236
237
backward([d, e], parallel_chunk_size=chunk_size)
238
239
- expected_jacobian = tensor_([2.0 * 3.0 * (a**2).item(), 2.0 * 4.0 * (a**3).item()])
+ J = tensor_([2.0 * 3.0 * (a**2).item(), 2.0 * 4.0 * (a**3).item()])
240
241
- assert_jac_close(a, expected_jacobian)
+ assert_jac_close(a, J)
242
243
244
def test_repeated_tensors():
0 commit comments