diff --git a/tests/influence/torch/test_influence_model.py b/tests/influence/torch/test_influence_model.py index 94374e796..529d1decd 100644 --- a/tests/influence/torch/test_influence_model.py +++ b/tests/influence/torch/test_influence_model.py @@ -854,13 +854,13 @@ def test_influences_cg( CgInfluence, maxiter=10, preconditioner=JacobiPreconditioner(), - use_block_cg=True, + solve_simultaneously=True, ), partial( CgInfluence, maxiter=10, preconditioner=NystroemPreconditioner(rank=5), - use_block_cg=True, + solve_simultaneously=True, ), ]