fix bug

wcxve · Nov 20, 2024 · 8d1f3d8 · 8d1f3d8
1 parent 8801f91
commit 8d1f3d8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/elisa/infer/results.py b/src/elisa/infer/results.py
@@ -642,7 +642,7 @@ def ci(
         else:
             rtol = jax.tree.map(float, dict(rtol))
             for k in rtol_keys:
-                rtol.set_default[k] = 1e-3
+                rtol.setdefault(k, 1e-3)
         if np.any([i > 0.1 for i in rtol.values()]):
             raise ValueError('rtol must be less than 0.1')
 

diff --git a/tests/test_results.py b/tests/test_results.py
@@ -94,10 +94,12 @@ def test_mle_ci_fn(mle_result2, powerlaw_flux):
         params['PowerLaw.alpha'], params['PowerLaw.K'], emin, emax
     )
 
+    ci0 = result.ci(params=[], fn={'fn': fn}, rtol={'fn': 5e-4})
     ci1 = result.ci(params=[], fn={'fn': fn})
     ci2 = result.ci(params=[], fn={'fn': fn}, method='boot')
     ci3 = result.ci(params=[], fn={'fn': fn}, method='boot', parallel=False)
     assert np.allclose(ci1.mle['fn'], fn_mle)
+    assert np.allclose(ci1.errors['fn'], ci0.errors['fn'])
     assert np.allclose(ci1.errors['fn'], ci2.errors['fn'], rtol=5e-2, atol=0.0)
     assert np.allclose(ci2.errors['fn'], ci3.errors['fn'])