Skip to content

Commit 26686b0

Browse files
committed
Merge pull request #121 from openml/hotfix/class-label
runs.run_task now works if the target feature the attribute name 'class' is capitalized.
2 parents 585e09a + 6bcb677 commit 26686b0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

openml/datasets/dataset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,8 @@ def _retrieve_class_labels(self):
249249
dataAttributes = dict(arffData['attributes'])
250250
if('class' in dataAttributes):
251251
return dataAttributes['class']
252+
elif('Class' in dataAttributes):
253+
return dataAttributes['Class']
252254
else:
253255
return None
254256

0 commit comments

Comments
 (0)