test(autogram): Remove batch_dim param when unused (#455)

Author: ValerianRey

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