pymc icon indicating copy to clipboard operation
pymc copied to clipboard
verified publisher icon pymc-devs

Implement nutpie as an external sampler

Open twiecki opened this issue 9 months ago • 3 comments

Summary

  • Fix parameter name mapping in NutPie integration (max_tree_depth → maxdepth)
  • Correctly handle InferenceData object returned by NutPie
  • Fix error in coords_and_dims_for_inferencedata function call
  • Handle progressbar parameter correctly
  • Simplify convergence checks to handle NutPie's different structure

Test plan

  • Run tests in tests/step_methods/test_external.py to verify NutPie integration works
  • Fixed parameter naming bugs with NutPie API

🤖 Generated with Claude Code

📚 Documentation preview 📚: https://pymc--7719.org.readthedocs.build/en/7719/

twiecki avatar Mar 10 '25 09:03 twiecki

@ricardov94

twiecki avatar Mar 22 '25 05:03 twiecki

@ricardoV94

It's on my stack :)

ricardoV94 avatar Mar 22 '25 14:03 ricardoV94

The new test file would have to be added to the yaml file (see failing pre-commit), and I'm sure they are failing looking at it

ricardoV94 avatar Apr 09 '25 12:04 ricardoV94

Superseded by #7880

ricardoV94 avatar Jul 30 '25 20:07 ricardoV94