minor jax tolerance bug fix

Author: sallen12

tests/test_crps.py

@@ -67,8 +67,8 @@ def test_crps_ensemble_corr(backend):
     res_pwm = sr.crps_ensemble(obs, fct, estimator="pwm", backend=backend)
     res_qd = sr.crps_ensemble(obs, fct, estimator="qd", backend=backend)
     if backend == "torch":
-        assert np.allclose(res_nrg, res_pwm, rtol=1e-04)
-        assert np.allclose(res_nrg, res_qd, rtol=1e-04)
+        assert np.allclose(res_nrg, res_pwm, rtol=1e-03)
+        assert np.allclose(res_nrg, res_qd, rtol=1e-03)
     else:
         assert np.allclose(res_nrg, res_pwm)
         assert np.allclose(res_nrg, res_qd)
@@ -78,8 +78,8 @@ def test_crps_ensemble_corr(backend):
     res_pwm = sr.crps_ensemble(obs, fct, ens_w=w, estimator="pwm", backend=backend)
     res_qd = sr.crps_ensemble(obs, fct, ens_w=w, estimator="qd", backend=backend)
     if backend == "torch":
-        assert np.allclose(res_nrg, res_pwm, rtol=1e-04)
-        assert np.allclose(res_nrg, res_qd, rtol=1e-04)
+        assert np.allclose(res_nrg, res_pwm, rtol=1e-03)
+        assert np.allclose(res_nrg, res_qd, rtol=1e-03)
     else:
         assert np.allclose(res_nrg, res_pwm)
         assert np.allclose(res_nrg, res_qd)