Fix flaky RScorer test (#876)

Increase sample size to reduce errors Signed-off-by: kgao <[email protected]> Signed-off-by: Gabriel Daiha <[email protected]>
py-why · May 10, 2024 · c0c5e6d · c0c5e6d
1 parent b66bc44
commit c0c5e6d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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):