def METHOD_NAME(theta_dict, design, observation_labels, target_labels):
    theta = torch.cat(list(theta_dict.values()), dim=-1)
    a, b = pyro.param("a", torch.tensor(0.0)), pyro.param("b", torch.tensor(0.0))
    pyro.sample("y", dist.Bernoulli(logits=a + b * theta))