Skip to content

doc(gh-2187): docstrings for Dirichlet and Geometric - #2227

Merged
Qazalbash merged 9 commits into
pyro-ppl:masterfrom
kir-7:doc/gh-2187-dirichlet-geometric
Jul 25, 2026
Merged

doc(gh-2187): docstrings for Dirichlet and Geometric#2227
Qazalbash merged 9 commits into
pyro-ppl:masterfrom
kir-7:doc/gh-2187-dirichlet-geometric

Conversation

@kir-7

@kir-7 kir-7 commented Jul 22, 2026

Copy link
Copy Markdown
Contributor

Part of #2187.

Adds documentation for 2 more distributions from the docs checklist, follows similar structure as #2202

  • Dirichlet — class, __init__, sample, log_prob, mean, variance, entropy
  • GeometricProbs — class, __init__, sample, log_prob, logits, mean, variance, entropy
  • GeometricLogits — class, __init__, probs, sample, log_prob, logits, mean, variance, entropy
  • Geometric — function

Docstrings only, no logic changes.

Verified:

  • make lint
  • make format

Comment thread numpyro/distributions/continuous.py Outdated
Comment thread numpyro/distributions/continuous.py Outdated
@kir-7
kir-7 requested a review from Qazalbash July 22, 2026 14:56

@Qazalbash Qazalbash left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Overall, it looks good. I have proposed some changes; could you please take a look?

Comment thread numpyro/distributions/continuous.py Outdated
Comment thread numpyro/distributions/continuous.py Outdated
Comment thread numpyro/distributions/continuous.py Outdated
Comment thread numpyro/distributions/discrete.py Outdated
Comment thread numpyro/distributions/discrete.py Outdated
Comment thread numpyro/distributions/discrete.py Outdated
Comment thread numpyro/distributions/discrete.py Outdated
Comment thread numpyro/distributions/discrete.py
Comment thread numpyro/distributions/discrete.py Outdated
Comment thread numpyro/distributions/discrete.py Outdated
@kir-7

kir-7 commented Jul 23, 2026

Copy link
Copy Markdown
Contributor Author

Thanks for the review, I have addressed the changes in the latest commit (7d53c8e).

@Qazalbash

Qazalbash commented Jul 24, 2026

Copy link
Copy Markdown
Collaborator

Thank you, @kir-7. Please fix the last change requested in this comment, and then this PR is ready for merge.

@kir-7

kir-7 commented Jul 24, 2026

Copy link
Copy Markdown
Contributor Author

Thank you, I have addressed the requested change.

@kir-7
kir-7 requested a review from Qazalbash July 24, 2026 09:51
@Qazalbash

Copy link
Copy Markdown
Collaborator

Merge the master branch after #2228 merges into it, so CI can pass.

@Qazalbash Qazalbash self-assigned this Jul 24, 2026
@Qazalbash
Qazalbash merged commit ca62e5b into pyro-ppl:master Jul 25, 2026
9 checks passed
@kir-7
kir-7 deleted the doc/gh-2187-dirichlet-geometric branch July 26, 2026 15:12
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.

3 participants