Skip to content

Commit 431cc1a

Browse files
committed
Type hint. Appveyor test.
1 parent 3937080 commit 431cc1a

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

openml/tasks/functions.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919
OpenMLLearningCurveTask,
2020
TaskTypeEnum,
2121
OpenMLRegressionTask,
22-
OpenMLSupervisedTask
22+
OpenMLSupervisedTask,
23+
OpenMLTask
2324
)
2425
import openml.utils
2526
import openml._api_calls
@@ -299,7 +300,7 @@ def get_tasks(task_ids, download_data=True):
299300
return tasks
300301

301302

302-
def get_task(task_id, download_data=True):
303+
def get_task(task_id: int, download_data: bool = True) -> OpenMLTask:
303304
"""Download OpenML task for a given task ID.
304305
305306
Downloads the task representation, while the data splits can be

0 commit comments

Comments
 (0)