Skip to content

Commit 4719013

Browse files
committed
Revert class labels
1 parent 959d56b commit 4719013

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

tests/test_datasets/test_dataset_functions.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -522,20 +522,18 @@ def test_publish_dataset(self):
522522
def test__retrieve_class_labels(self):
523523
openml.config.set_root_cache_directory(self.static_cache_dir)
524524
labels = openml.datasets.get_dataset(2).retrieve_class_labels()
525-
#TODO diff in prod
526-
assert labels == ['nowin', 'won']
525+
assert labels == ["1", "2", "3", "4", "5", "U"]
527526

528527
labels = openml.datasets.get_dataset(2).retrieve_class_labels(
529528
target_name="product-type",
530529
)
531-
#TODO no product-type in test/local
532-
assert labels == None
530+
assert labels == ["C", "H", "G"]
533531

534532
# Test workaround for string-typed class labels
535533
custom_ds = openml.datasets.get_dataset(2)
536534
custom_ds.features[31].data_type = "string"
537535
labels = custom_ds.retrieve_class_labels(target_name=custom_ds.features[31].name)
538-
assert labels == ["f","t"]
536+
assert labels == ["COIL", "SHEET"]
539537

540538
@pytest.mark.test_server()
541539
def test_upload_dataset_with_url(self):

0 commit comments

Comments
 (0)