Fix fixture usage in test

aai-institute · Jun 5, 2024 · 5b34406 · 5b34406
1 parent f25f289
commit 5b34406
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/influence/torch/test_batch_operation.py b/tests/influence/torch/test_batch_operation.py
@@ -118,7 +118,7 @@ def compute_gn_mat_analytical(self):
         [(astuple(tp.model_params), 1e-3) for tp in test_parameters],
         indirect=["model_data"],
     )
-    def test_analytical_comparison(self, tol):
+    def test_analytical_comparison(self, model_data, tol):
         gn_autograd = self.gn_op.apply(TorchBatch(self.x, self.y), self.vec)
         gn_autograd_dict = self.gn_op.apply_to_dict(
             TorchBatch(self.x, self.y), align_structure(self.params, self.vec)
@@ -136,7 +136,7 @@ def test_analytical_comparison(self, tol):
         [(astuple(tp.model_params), 1e-3) for tp in test_parameters],
         indirect=["model_data"],
     )
-    def test_flattening_commutation(self, tol):
+    def test_flattening_commutation(self, model_data, tol):
         batch_size = 10
         rand_mat_dict = {
             k: torch.randn(batch_size, *t.shape) for k, t in self.params.items()