ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-05-29 09:38:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for setting default value to auto connected argparse arguments

Browse Source
This commit is contained in:
allegroai 2021-10-16 23:05:45 +03:00
parent 9085847c6d
commit 0bca84375b
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clearml/backend_interface/task/args.py
View File

@ -179,15 +179,16 @@ class _Arguments(object):
# Skip excluded arguments, Add prefix.
if Session.check_min_api_version('2.9'):
include_all = self._exclude_parser_args.get("*", True)
task_defaults = dict(
[(self._prefix_args + k, v) for k, v in task_defaults.items()
if self._exclude_parser_args.get(k, True)])
if self._exclude_parser_args.get(k, include_all)])
task_defaults_descriptions = dict(
[(self._prefix_args + k, v) for k, v in task_defaults_descriptions.items()
if self._exclude_parser_args.get(k, True)])
if self._exclude_parser_args.get(k, include_all)])
task_defaults_types = dict(
[(self._prefix_args + k, v) for k, v in task_defaults_types.items()
if self._exclude_parser_args.get(k, True)])
if self._exclude_parser_args.get(k, include_all)])
else:
task_defaults = dict(
[(k, v) for k, v in task_defaults.items() if self._exclude_parser_args.get(k, True)])

8
clearml/task.py
View File

@ -328,7 +328,9 @@ class Task(_Task):
- A dictionary - In addition to a boolean, you can use a dictionary for fined grained control of connected
arguments. The dictionary keys are argparse variable names and the values are booleans.
The ``False`` value excludes the specified argument from the Task's parameter section.
Keys missing from the dictionary default to ``True``, and an empty dictionary defaults to ``False``.
Keys missing from the dictionary default to ``True``, you can change it to be ``False`` by adding
``*`` key as ``False`` to the dictionary.
An empty dictionary defaults to ``False``.
For example:
@ -336,6 +338,10 @@ class Task(_Task):
auto_connect_arg_parser={'do_not_include_me': False, }
.. code-block:: py
auto_connect_arg_parser={"only_include_me": True, "*": False}
.. note::
To manually connect an argparse, use :meth:`Task.connect`.