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

Task.init argument auto_connect_arg_parser can accept a dictionary disabling specific keys from the argparser

Browse Source
This commit is contained in:
allegroai
2020-05-13 20:42:33 +03:00
parent d2c9523769
commit cb8887da72
2 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
trains/backend_interface/task/args.py
View File

@@ -40,6 +40,10 @@ class _Arguments(object):
def __init__(self, task):
super(_Arguments, self).__init__()
self._task = task
self._exclude_parser_args = {}
def exclude_parser_args(self, excluded_args):
self._exclude_parser_args = excluded_args or {}
def set_defaults(self, *dicts, **kwargs):
self._task.set_parameters(*dicts, **kwargs)
@@ -126,9 +130,11 @@ class _Arguments(object):
task_defaults[k] = str(v)
except Exception:
del task_defaults[k]
# Add prefix, TODO: add argparse prefix
# task_defaults = dict([(self._prefix_args + k, v) for k, v in task_defaults.items()])
task_defaults = dict([(k, v) for k, v in task_defaults.items()])
# Skip excluded arguments, Add prefix, TODO: add argparse prefix
# task_defaults = dict([(self._prefix_args + k, v) for k, v in task_defaults.items()
# if k not in self._exclude_parser_args])
task_defaults = dict([(k, v) for k, v in task_defaults.items() if self._exclude_parser_args.get(k, True)])
# Store to task
self._task.update_parameters(task_defaults)
@@ -154,7 +160,7 @@ class _Arguments(object):
# task_arguments = dict([(k[len(self._prefix_args):], v) for k, v in self._task.get_parameters().items()
# if k.startswith(self._prefix_args)])
task_arguments = dict([(k, v) for k, v in self._task.get_parameters().items()
if not k.startswith(self._prefix_tf_defines)])
if not k.startswith(self._prefix_tf_defines) and self._exclude_parser_args.get(k, True)])
arg_parser_argeuments = {}
for k, v in task_arguments.items():
# python2 unicode support