From cb31ee2bf5610faa7b815acf719cd48975823756 Mon Sep 17 00:00:00 2001 From: Alexander Fischer Date: Wed, 8 Jan 2025 22:06:28 +0100 Subject: [PATCH] add jax demeaning integration tests (#783) --- tests/test_vs_fixest.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/test_vs_fixest.py b/tests/test_vs_fixest.py index 69a5ff3c..b317b682 100644 --- a/tests/test_vs_fixest.py +++ b/tests/test_vs_fixest.py @@ -198,6 +198,7 @@ def check_relative_diff(x1, x2, tol, msg=None): @pytest.mark.parametrize("fml", ols_fmls + ols_but_not_poisson_fml) @pytest.mark.parametrize("adj", [True]) @pytest.mark.parametrize("cluster_adj", [True]) +@pytest.mark.parametrize("demeaner_backend", ["numba", "jax"]) def test_single_fit_feols( data_feols, dropna, @@ -207,6 +208,7 @@ def test_single_fit_feols( fml, adj, cluster_adj, + demeaner_backend, ): global test_counter_feols test_counter_feols += 1 @@ -234,7 +236,14 @@ def test_single_fit_feols( r_inference = _get_r_inference(inference) - mod = pf.feols(fml=fml, data=data, vcov=inference, weights=weights, ssc=ssc_) + mod = pf.feols( + fml=fml, + data=data, + vcov=inference, + weights=weights, + ssc=ssc_, + demeaner_backend=demeaner_backend, + ) if weights is not None: r_fixest = fixest.feols( ro.Formula(r_fml),