Fixing generation of class ids for only one class
Summary
This MR introduces the following changes
- Fixes an issue that when only one class is given via
class_labels
(for plotting reasons or so), that anint
was returned and not alist
. This would kill the loader functions because they loop over this output. This should fix the issue.