Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make figure in AR2 example reproducible #1816

Merged
merged 6 commits into from
Jun 23, 2024

Conversation

damonbayer
Copy link
Contributor

This PR adds back in code to produce the ar2 plot in the ar2 example. This is based on the code that was removed in #1350. It was last available in https://github.com/pyro-ppl/numpyro/blob/c5b1a9229daf964a98eb12a8ad905e0d0a6c9437/examples/ar2.py.

The figure was generated with:

python ar2.py --num-samples 1000 --num-warmup 1000 --num-chains 4

Closes #1815.

@damonbayer damonbayer changed the title Make ar2 plot reproducible Make figure in AR2 example reproducible Jun 18, 2024
Copy link
Member

@fehiepsi fehiepsi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, @damonbayer!

@damonbayer
Copy link
Contributor Author

CI checks should now pass. Sorry about that.

examples/ar2.py Outdated
@@ -64,8 +64,6 @@ def transition(carry, _):
with numpyro.handlers.condition(data={"y": y[2:]}):
_, mu = scan(transition, init, timesteps)

numpyro.deterministic("mu", mu)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this intended?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No 🤦‍♂️

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed in 8ebbaa7

@damonbayer
Copy link
Contributor Author

@fehiepsi I don't think any of the test failures are related to my contribution. Please let me know if you disagree.

@fehiepsi
Copy link
Member

Yes they are unrelated - @juanitorduz is working on them in #1817

@fehiepsi
Copy link
Member

@damonbayer Could you sync with master? They are fixed upstream.

@damonbayer
Copy link
Contributor Author

@fehiepsi Fork is synced!

@fehiepsi fehiepsi merged commit 491a0cd into pyro-ppl:master Jun 23, 2024
4 checks passed
@damonbayer damonbayer deleted the dmb_ar2_plot branch June 23, 2024 11:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Figure in AR2 example is not reproducible
2 participants