From e601d5bbbc08ed497f1355e5a155d05d1ad9ae20 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Val=C3=A9rian=20Rey?= Date: Mon, 13 Oct 2025 20:06:14 +0200 Subject: [PATCH] test(autogram): Remove batch_dim param when unused --- tests/unit/autogram/test_engine.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/unit/autogram/test_engine.py b/tests/unit/autogram/test_engine.py index c98831dd..933154e1 100644 --- a/tests/unit/autogram/test_engine.py +++ b/tests/unit/autogram/test_engine.py @@ -468,8 +468,7 @@ def test_compute_gramian_manual(): [1], ], ) -@mark.parametrize("batch_dim", [0, None]) -def test_reshape_equivariance(shape: list[int], batch_dim: int | None): +def test_reshape_equivariance(shape: list[int]): """ Test equivariance of `compute_gramian` under reshape operation. More precisely, if we reshape the `output` to some `shape`, then the result is the same as reshaping the Gramian to the