Skip to content

Commit ea46d0d

Browse files
authored
Fix flaky RScorer test (#876)
Increase sample size to reduce errors Signed-off-by: kgao <[email protected]>
1 parent db1e254 commit ea46d0d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

econml/tests/test_rscorer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@ def _fit_model(name, model, Y, T, X):
2121
class TestRScorer(unittest.TestCase):
2222

2323
def _get_data(self):
24-
X = np.random.normal(0, 1, size=(2000, 2))
25-
T = np.random.binomial(1, .5, size=(2000,))
26-
y = X[:, 0] * T + np.random.normal(size=(2000,))
24+
X = np.random.normal(0, 1, size=(100000, 2))
25+
T = np.random.binomial(1, .5, size=(100000,))
26+
y = X[:, 0] * T + np.random.normal(size=(100000,))
2727
return y, T, X, X[:, 0]
2828

2929
def test_comparison(self):

0 commit comments

Comments
 (0)