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