Conversation
There was a problem hiding this comment.
Code Review
This pull request updates the flowmc dependency to version 0.6.1 and replaces the custom _HMC implementation with the native HMC sampler from flowMC. As part of this migration, references to mass_matrix have been renamed to condition_matrix across the codebase, and a new Pydantic validator has been added. The review feedback highlights opportunities to improve robustness: validating the shape of condition_matrix against n_dims to prevent JAX shape mismatches, removing unreachable list-checking code in the Pydantic validator, and raising an error in diag_condition_matrix.py if any computed variance is zero or negative.
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>
Summary
Update dependencies and APIs for the latest flowMC v0.6.1
Related To
flowMC>=0.5.0#790