Conversation
There was a problem hiding this comment.
Code Review
This pull request wraps the trapezoidal integration result for the primary mass distribution normalization constant in a logarithm (jnp.log) to correctly compute _logZ. The reviewer notes that directly applying jnp.log without safety guards can lead to numerical instability, such as underflow to zero resulting in -inf or NaN values. They suggest adding a guard using jnp.where to handle non-positive or invalid integration results safely.
Important
The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
No description provided.