Skip to content

Conversation

fehiepsi
Copy link
Member

@fehiepsi fehiepsi commented Sep 5, 2024

Currently, enumerate_support of DiscreteUniform raises an error when either low or high parameter has non-trivial shape. This fixes the issue and adds a test to cover the change.

Unblock #1835

Deathn0t added a commit to Deathn0t/numpyro that referenced this pull request Sep 5, 2024
@martinjankowiak martinjankowiak merged commit ca8fb39 into pyro-ppl:master Sep 16, 2024
4 checks passed
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.

2 participants