fix initializer_kwargs test

theislab · Dec 10, 2024 · 3fd0fac · 3fd0fac
1 parent 39e6c03
commit 3fd0fac
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/problems/generic/test_sinkhorn_problem.py b/tests/problems/generic/test_sinkhorn_problem.py
@@ -160,9 +160,10 @@ def test_pass_arguments(self, adata_time: AnnData, args_to_check: Mapping[str, A
         solver = problem[(0, 1)].solver.solver
         args = sinkhorn_solver_args if args_to_check["rank"] == -1 else lr_sinkhorn_solver_args
         for arg, val in args.items():
-            assert hasattr(solver, val), val
-            el = getattr(solver, val)[0] if isinstance(getattr(solver, val), tuple) else getattr(solver, val)
-            assert el == args_to_check[arg], arg
+            if val != "initializer_kwargs":
+                assert hasattr(solver, val), val
+                el = getattr(solver, val)[0] if isinstance(getattr(solver, val), tuple) else getattr(solver, val)
+                assert el == args_to_check[arg], arg
 
         lin_prob = problem[(0, 1)]._solver._problem
         for arg, val in lin_prob_args.items():