Tighter conditions for loops in _poisson sampling (#583)
Tighter conditions for loops in _poisson sampling (#583)

* Tighter conditions for loops in _poisson sampling

fixes issue #582

* fixing linting issues
pyro-ppl/numpyro