Skip to content

Commit 022f5c1

Browse files
committed
MAINT work on Andreas' comments'
1 parent 26ac44e commit 022f5c1

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

openml/tasks/task.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def get_dataset(self):
3333
"""Download dataset associated with task"""
3434
return datasets.get_dataset(self.dataset_id)
3535

36-
def get_X_and_Y(self):
36+
def get_X_and_y(self):
3737
dataset = self.get_dataset()
3838
# Replace with retrieve from cache
3939
if 'Supervised Classification'.lower() in self.task_type.lower():
@@ -42,9 +42,9 @@ def get_X_and_Y(self):
4242
target_dtype = float
4343
else:
4444
raise NotImplementedError(self.task_type)
45-
X_and_Y = dataset.get_data(target=self.target_feature,
45+
X_and_y = dataset.get_data(target=self.target_feature,
4646
target_dtype=target_dtype)
47-
return X_and_Y
47+
return X_and_y
4848

4949
def get_train_test_split_indices(self, fold=0, repeat=0):
5050
# Replace with retrieve from cache

tests/datasets/test_datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_get_cached_dataset_description_not_cached(self):
5959

6060
def test_get_cached_dataset_arff(self):
6161
openml.config.set_cache_directory(self.static_cache_dir)
62-
description = openml.datasets.functions._get_cached_dataset_arff(2)
62+
description = openml.datasets.functions._get_cached_dataset_arff(did=2)
6363
self.assertIsInstance(description, str)
6464

6565
def test_get_cached_dataset_arff_not_cached(self):

tests/tasks/test_task.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def test_get_dataset(self, patch):
3232
def test_get_X_and_Y(self):
3333
# Classification task
3434
task = openml.tasks.get_task(1)
35-
X, Y = task.get_X_and_Y()
35+
X, Y = task.get_X_and_y()
3636
self.assertEqual((898, 38), X.shape)
3737
self.assertIsInstance(X, np.ndarray)
3838
self.assertEqual((898, ), Y.shape)
@@ -41,7 +41,7 @@ def test_get_X_and_Y(self):
4141

4242
# Regression task
4343
task = openml.tasks.get_task(2280)
44-
X, Y = task.get_X_and_Y()
44+
X, Y = task.get_X_and_y()
4545
self.assertEqual((8192, 8), X.shape)
4646
self.assertIsInstance(X, np.ndarray)
4747
self.assertEqual((8192,), Y.shape)

0 commit comments

Comments
 (0)