import pyhf

spec = {
    "channels": [
        {
            "name": "SR",
            "samples": [
                {
                    "data": [25.0],
                    "modifiers": [
                        {"data": None, "name": "mu", "type": "normfactor"},
                    ],
                    "name": "Signal",
                },
            ],
        }
    ],
    "measurements": [
        {"config": {"parameters": [], "poi": "mu"}, "name": "minimal_example"}
    ],
    "observations": [{"data": [35.0], "name": "SR"}],
    "version": "1.0.0",
}


def fit(spec):
    ws = pyhf.Workspace(spec)
    model = ws.model()
    data = ws.data(model) + model.config.auxdata
    print(pyhf.infer.mle.fit(data, model))


# fit with one modifier
fit(spec)

# add a second POI
spec["channels"][0]["samples"][0]["modifiers"].append(
    {"data": None, "name": "mu", "type": "normfactor"}
)
fit(spec)