🐛 plot Calibration error

[mrynik]

Hey, just spotted a bug

torch-uncertainty/torch_uncertainty/metrics/classification/calibration_error.py

Line 63 in ff355d9

fig.tight_layout()

This will fail if you pass in the ax object to the _ce_plot function because in that case fig is initialized to None.

torch-uncertainty/torch_uncertainty/metrics/classification/calibration_error.py

Line 18 in ff355d9

fig, ax = plt.subplots(figsize=(6, 6)) if ax is None else (None, ax)

[o-laurent]

Hi mrynik!

Thank you for the pointer! I pushed a quick fix on dev. We intend to rework the plots a bit in the following releases (#54)!