Skip to content

Commit 03d7807

Browse files
committed
small update to run functions related to tags
1 parent df9f7bc commit 03d7807

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

openml/runs/functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323

2424

25-
def run_task(task, model, avoid_duplicate_runs=True):
25+
def run_task(task, model, avoid_duplicate_runs=True, flow_tags=None):
2626
"""Performs a CV run on the dataset of the given task, using the split.
2727
2828
Parameters
@@ -70,7 +70,7 @@ def run_task(task, model, avoid_duplicate_runs=True):
7070
run_environment = _get_version_information()
7171
tags = ['openml-python', run_environment[1]]
7272
# execute the run
73-
run = OpenMLRun(task_id=task.task_id, flow_id=None, dataset_id=dataset.dataset_id, model=model)
73+
run = OpenMLRun(task_id=task.task_id, flow_id=None, dataset_id=dataset.dataset_id, model=model, tags=tags)
7474
run.data_content, run.trace_content, run.trace_attributes = _run_task_get_arffcontent(model, task, class_labels)
7575

7676

0 commit comments

Comments
 (0)