diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py index de9218dc..0bced2f2 100644 --- a/tests/test_optimizers.py +++ b/tests/test_optimizers.py @@ -707,7 +707,13 @@ def test_trac_optimizer_erf_imag(): 'params', [ {'merge_dims': True, 'precondition_1d': True, 'max_precondition_dim': 4, 'precondition_frequency': 1}, - {'merge_dims': True, 'precondition_1d': False, 'max_precondition_dim': 1, 'precondition_frequency': 1}, + { + 'merge_dims': True, + 'precondition_1d': False, + 'max_precondition_dim': 1, + 'precondition_frequency': 1, + 'normalize_gradient': True, + }, ], ) def test_soap_parameters(params, environment):