From c0c5e6d1a1ddfb371dd642b5c255379cdcf2e584 Mon Sep 17 00:00:00 2001 From: Kevin Gao Date: Tue, 7 May 2024 14:24:43 -0400 Subject: [PATCH] Fix flaky RScorer test (#876) Increase sample size to reduce errors Signed-off-by: kgao Signed-off-by: Gabriel Daiha --- econml/tests/test_rscorer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/econml/tests/test_rscorer.py b/econml/tests/test_rscorer.py index 9ca2d5c50..4f41b5016 100644 --- a/econml/tests/test_rscorer.py +++ b/econml/tests/test_rscorer.py @@ -21,9 +21,9 @@ def _fit_model(name, model, Y, T, X): class TestRScorer(unittest.TestCase): def _get_data(self): - X = np.random.normal(0, 1, size=(2000, 2)) - T = np.random.binomial(1, .5, size=(2000,)) - y = X[:, 0] * T + np.random.normal(size=(2000,)) + X = np.random.normal(0, 1, size=(100000, 2)) + T = np.random.binomial(1, .5, size=(100000,)) + y = X[:, 0] * T + np.random.normal(size=(100000,)) return y, T, X, X[:, 0] def test_comparison(self):