def METHOD_NAME(subsample_size):
    size = 10
    def model():
        with pyro.plate("J", size, subsample_size=subsample_size):
            pyro.sample("x", dist.Normal(0, 1))
    kernel = NUTS(model)
    mcmc = MCMC(kernel, 10)
    if subsample_size < size:
        with pytest.raises(RuntimeError, match="subsample"):
            mcmc.run()
    else:
        mcmc.run()