zclaytor / kiauhoku

Python utilities for stellar model grid interpolation
MIT License
16 stars 6 forks source link

Enable parallelization in `StarGridInterpolator.mcmc_star` #9

Open zclaytor opened 2 months ago

zclaytor commented 2 months ago

The mcmc_star method is a wrapper for the emcee ensemble sampler, which allows parallelization using multithreading. However, the current implementation in kiauhoku doesn't allow the user to pass the parallelization keywords to the ensemble sampler. This feature request is to allow the user to pass any valid EnsembleSampler keywords through mcmc_star to fully utilize the power of emcee.

Use case would look something like:

grid.mcmc_star(
    # regular inputs to mcmc_star
    emcee_kwargs={"nthreads": ...} # or whatever emcee calls a thread
)