fix datapath bart data

[juanitorduz]

I was trying to simply run

from pyro.contrib.examples.bart import load_bart_od

dataset = load_bart_od()

and I was getting a timeout error. I saw this in NumPyro and this was fixed in pyro-ppl/numpyro#1996 by @fehiepsi . I tested the change locally (same strategy) and works as expected 🙏 .

[fritzo]

Thanks for the fix! LGTM except you'll need to make lint or black or something. Looks like your editor accidentally reformatted some lines to exceed this codebase's max line length.

[juanitorduz]

Fixed in e385602 (and ba6b5b0) :)