Skip to content

Commit fc22729

Browse files
committed
add random seed argument
Signed-off-by: Nathaniel <NathanielF@users.noreply.github.com>
1 parent 8328ebe commit fc22729

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

causalpy/tests/test_pymc_experiments.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,13 @@ def test_regression_kink_gradient_change():
6363

6464
def test_inverse_prop():
6565
df = cp.load_data("nhefs")
66-
sample_kwargs = {"tune": 100, "draws": 500, "chains": 2, "cores": 2}
66+
sample_kwargs = {
67+
"tune": 100,
68+
"draws": 500,
69+
"chains": 2,
70+
"cores": 2,
71+
"random_seed": 100,
72+
}
6773
result = cp.pymc_experiments.InversePropensityWeighting(
6874
df,
6975
formula="trt ~ 1 + age + race",

0 commit comments

Comments
 (0)