Merge branch 'main' into kebatt/updateCI

py-why · May 14, 2024 · ab90fb1 · ab90fb1
2 parents 07eb2e3 + ea46d0d
commit ab90fb1
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):