-
Notifications
You must be signed in to change notification settings - Fork 151
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
Fixes current density estimator bug #1155
Fixes current density estimator bug #1155
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #1155 +/- ##
==========================================
- Coverage 83.93% 74.23% -9.71%
==========================================
Files 90 90
Lines 6930 6931 +1
==========================================
- Hits 5817 5145 -672
- Misses 1113 1786 +673
Flags with carried forward coverage won't be shown. Click here to find out more.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks a ton!
*sample_shape, | ||
condition_batch_dim, | ||
) | ||
+ self.input_shape |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
thanks! Would
(*sample_shape, condition_batch_dim, *self.input_shape)
also work? It's just a bit cleaner IMO.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Jup, fixed.
Added tests that check if batched log_prob/sample will be the same as unbatched. Will merge after tests passed. |
What does this implement/fix? Explain your changes
Fixes #1154
TODO