Add a tolerance to differential_evolution

PrincetonUniversity · Oct 15, 2024 · 3a0cdbe · 3a0cdbe
1 parent ecfb6f2
commit 3a0cdbe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/tests/composition/pec/test_parameterestimationcomposition.py b/tests/composition/pec/test_parameterestimationcomposition.py
@@ -289,7 +289,12 @@ def reward_rate(sim_data):
         pec.run(inputs={comp: trial_inputs})
 
     if expected_result is not None:
-        np.testing.assert_allclose(list(pec.optimized_parameter_values.values()), expected_result)
+        if opt_method == "differential_evolution":
+            np.testing.assert_allclose(
+                list(pec.optimized_parameter_values.values()), expected_result, rtol=1e-3
+            )
+        else:
+            np.testing.assert_allclose(list(pec.optimized_parameter_values.values()), expected_result)
 
 
 def test_parameter_estimation_ddm_cond(func_mode):

diff --git a/tests/composition/pec/test_stab_flex_pec_fit.py b/tests/composition/pec/test_stab_flex_pec_fit.py
@@ -480,7 +480,7 @@ def test_stab_flex_cond_fit():
 
     inputs = make_input_dict(comp, taskTrain, stimulusTrain, cueTrain, correctResponse)
 
-    ret = pec.run(inputs=inputs)
+    pec.run(inputs=inputs)
     optimal_parameters = pec.optimized_parameter_values
 
     # These aren't the recovered parameters, we are doing too few trials and too few estimates to get the correct