ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-08 23:57:16 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add argparse argument passing to Task.create() (issue #230)

Browse Source
This commit is contained in:
allegroai 2020-12-30 16:38:56 +02:00
parent 6ca4836bea
commit 2b81bc33a5
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/task.py
View File

@ -594,6 +594,7 @@ class Task(_Task):
packages=None, # Optional[Sequence[str]] packages=None, # Optional[Sequence[str]]
requirements_file=None, # Optional[Union[str, Path]] requirements_file=None, # Optional[Union[str, Path]]
docker=None, # Optional[str] docker=None, # Optional[str]
argparse_args=None, # Optional[Sequence[Tuple[str, str]]]
base_task_id=None, # Optional[str] base_task_id=None, # Optional[str]
add_task_init_call=True, # bool add_task_init_call=True, # bool
): ):
@ -627,6 +628,8 @@ class Task(_Task):
:param requirements_file: Specify requirements.txt file to install when setting the session. :param requirements_file: Specify requirements.txt file to install when setting the session.
If not provided, the requirements.txt from the repository will be used. If not provided, the requirements.txt from the repository will be used.
:param docker: Select the docker image to be executed in by the remote session :param docker: Select the docker image to be executed in by the remote session
:param argparse_args: Arguments to pass to the remote execution, list of string pairs (argument, value)
Notice, only supported if the codebase itself uses argparse.ArgumentParser
:param base_task_id: Use a pre-existing task in the system, instead of a local repo/script. :param base_task_id: Use a pre-existing task in the system, instead of a local repo/script.
Essentially clones an existing task and overrides arguments/requirements. Essentially clones an existing task and overrides arguments/requirements.
:param add_task_init_call: If True, a 'Task.init()' call is added to the script entry point in remote execution. :param add_task_init_call: If True, a 'Task.init()' call is added to the script entry point in remote execution.
@ -650,6 +653,9 @@ class Task(_Task):
raise_on_missing_entries=False, raise_on_missing_entries=False,
) )
task = manual_populate.create_task() task = manual_populate.create_task()
if task and argparse_args:
manual_populate.update_task_args(argparse_args)
task.reload()
return task return task