emaballarin/nmc-numpyro

Unwrap parameters from ParamInfo before returning | Avoid ParamInfo altogether

emaballarin opened this issue 3 years ago · 0 comments

emaballarin commented 3 years ago

See:

nmc-numpyro/nmc_numpyro/nmc.py

Line 97 in 20dd6c1

if not isinstance(init_params, ParamInfo):
nmc-numpyro/nmc_numpyro/nmc.py

Line 98 in 20dd6c1

init_params = ParamInfo(init_params)
nmc-numpyro/nmc_numpyro/nmc.py

Line 175 in 20dd6c1

ParamInfo(proposed_state),

As per title.