Skip to content

Commit 29c1e29

Browse files
committed
Increase prior predictive draws to 100 in notebook runner
Updated the mock for pm.sample() to use 100 draws instead of 50 for prior predictive sampling, as reflected in both the injected script and documentation. This change aims to provide more robust validation during notebook execution.
1 parent 702089e commit 29c1e29

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

scripts/run_notebooks/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ This script runs Jupyter notebooks from `docs/source/notebooks/` to validate the
44

55
## How It Works
66

7-
1. **Mocks `pm.sample()`** — Replaces MCMC sampling with prior predictive (1 chain × 50 draws) for speed
7+
1. **Mocks `pm.sample()`** — Replaces MCMC sampling with prior predictive (1 chain × 100 draws) for speed
88
2. **Uses Papermill** — Executes notebooks programmatically
99
3. **Discards outputs** — Only checks for errors, doesn't save results
1010

scripts/run_notebooks/injected.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def mock_sample(*args, **kwargs):
1515
first_arg = args[0]
1616
if isinstance(first_arg, pm.Model):
1717
model = first_arg
18-
n_draws = 50
18+
n_draws = 100
1919

2020
idata = pm.sample_prior_predictive(
2121
model=model,

0 commit comments

Comments
 (0)