Fix in case --project is not provided

This commit is contained in:
allegroai 2021-06-03 08:49:41 +03:00
parent 9b81d5e02d
commit 925407923f

View File

@ -39,7 +39,7 @@ def cmd_triton(args):
if not args.id and not args.name: if not args.id and not args.name:
raise ValueError("Serving service must have a name, use --name <service_name>") raise ValueError("Serving service must have a name, use --name <service_name>")
if args.cmd_id or (args.id and not args.project and not args.name): if args.cmd_id or (args.id and not args.name):
a_serving = ServingService(task_id=args.cmd_id or args.id) a_serving = ServingService(task_id=args.cmd_id or args.id)
else: else:
a_serving = ServingService(task_project=args.project, task_name=args.name, engine_type='triton') a_serving = ServingService(task_project=args.project, task_name=args.name, engine_type='triton')