The issue is related to: #20 (comment)
We are currently returning label_feature + label for each dataset and some datasets lack this label_feature. Now we are returning label if label_feature does not exist.
This should be tied up with the PR implementing the callable that helps creating the labels.