update: test_soap_parameters

kozistr · Sep 29, 2024 · 851eb0d · 851eb0d
1 parent cc3b2df
commit 851eb0d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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):