[JOSS REVIEW] "More detailed intro" example throws an error

[aneeshnaik]

This issue is part of the JOSS review for sbijax. Review thread here.

The "more detailed intro" example in the docs throws an error. In the line model = nle(prior, neural_network), I get

TypeError                                 Traceback (most recent call last)
Cell In[13], line 2
      1 neural_network = make_custom_affine_maf(n_dim_data, n_layers, hidden_sizes)
----> 2 model = nle(prior, neural_network)

TypeError: nle() takes 1 positional argument but 2 were given

This is with sbijax 0.4.0