Fix passing --id and --project/--name

This commit is contained in:
allegroai 2021-06-02 14:18:39 +03:00
parent 3287e4e4ad
commit 9b81d5e02d
2 changed files with 15 additions and 7 deletions

View File

@ -13,7 +13,9 @@ def restore_state(args):
state = json.load(f)
except Exception:
state = {}
# store command line passed ID
args.cmd_id = getattr(args, 'id', None)
# restore ID from state
args.id = getattr(args, 'id', None) or state.get('id')
return args
@ -37,8 +39,8 @@ def cmd_triton(args):
if not args.id and not args.name:
raise ValueError("Serving service must have a name, use --name <service_name>")
if args.id and not args.project and not args.name:
a_serving = ServingService(task_id=args.id)
if args.cmd_id or (args.id and not args.project and not args.name):
a_serving = ServingService(task_id=args.cmd_id or args.id)
else:
a_serving = ServingService(task_project=args.project, task_name=args.name, engine_type='triton')
args.id = a_serving.get_id()

View File

@ -45,10 +45,16 @@ class ServingService(object):
elif task_id:
self._task = Task.get_task(task_id=task_id)
else:
# noinspection PyProtectedMember
if Task._query_tasks(project_name=task_project, task_name=task_name):
self._task = Task.get_task(project_name=task_project, task_name=task_name)
else:
# try to get a Task if we can find one
self._task = None
try:
# noinspection PyProtectedMember
if Task._query_tasks(project_name=task_project, task_name=task_name):
self._task = Task.get_task(project_name=task_project, task_name=task_name)
except ValueError:
pass
if not self._task:
self._task = Task.create(
project_name=task_project, task_name=task_name, task_type=Task.TaskTypes.service,
repo="https://github.com/allegroai/clearml-serving.git",