CmdStanPy helpers

cmdstanpy_helpers.py

from pathlib import Path
import cmdstanpy
import ujson

def stanc(name, code, workdir=".stan"):
    Path(workdir).mkdir(exist_ok=True)
    path = Path(workdir) / f"{name}.stan"
    with open(path, "w") as f:
        f.write(code)
    model = cmdstanpy.CmdStanModel(stan_file=path, stanc_options={'warn-pedantic': True})
    return model

def tojson(path, **kwargs):
    out = {}
    for k, v in kwargs.items():
        if isinstance(v, np.ndarray):
            v = v.tolist()
        # print(k, v)
        out[k] = v
    with open(path, "w") as f:
        ujson.dump(out, f)
    return path