Make diagnostic plots work with one-parameter models

stefanradev93 / BayesFlow

A Python library for amortized Bayesian workflows using generative neural networks.

https://bayesflow.org/

MIT License

346 stars 50 forks source link

Make diagnostic plots work with one-parameter models #137

Closed Kucharssim closed 7 months ago

Kucharssim commented 7 months ago

fixes https://github.com/stefanradev93/BayesFlow/issues/135

Just ensuring that we can loop over ax even when it's returned by plt.subplots(nrows=1, ncols=1). There is some code duplication between different plots, but I just made a simple fix where needed; it doesn't seem that bad to warrant a larger refactor just yet 🙂

stefanradev93 commented 7 months ago

Thanks, Simon! No worries, we are currently refactoring the backend massively.

@LarsKue Tests seem to still be failing. Any idea why?

LarsKue commented 7 months ago

Most likely the same issue as detailed in #133 .