@@ -165,7 +165,6 @@ def add_deployment(args):
165
165
ret_json = rfapi .add_deployment (
166
166
args .api_key ,
167
167
args .security_level ,
168
- args .cloud_provider ,
169
168
args .machine_type ,
170
169
args .deployment_name ,
171
170
args .inference_version ,
@@ -500,29 +499,29 @@ def _add_deployment_parser(subparsers):
500
499
deployment_delete_parser = deployment_subparsers .add_parser ("delete" , help = "delete a dedicated deployment" )
501
500
502
501
deployment_add_parser .set_defaults (func = add_deployment )
503
- deployment_add_parser .add_argument ("api_key" , help = "api key" )
504
- deployment_add_parser .add_argument ("security_level" , help = "security level (protected)" , default = "protected" )
505
- deployment_add_parser .add_argument ("cloud_provider" , help = "cloud provider (gcp)" , default = "gcp" )
502
+ deployment_add_parser .add_argument ("-a" , dest = "api_key" , help = "api key" )
503
+ deployment_add_parser .add_argument ("-s" , dest = "security_level" , help = "security level (protected)" , default = "protected" )
506
504
deployment_add_parser .add_argument (
507
- "machine_type" ,
508
- help = "machine type (gcp-n1-standard-4-gpu, gcp-g2-standard-8-gpu)" ,
509
- default = "gcp-n1-standard-4-gpu" ,
505
+ "-m" ,
506
+ dest = "machine_type" ,
507
+ help = "machine type (gcp-n2-cpu, gcp-t4-gpu, gcp-l4-gpu)" ,
508
+ default = "gcp-n2-cpu" ,
510
509
)
511
- deployment_add_parser .add_argument ("deployment_name" , help = "deployment name" )
510
+ deployment_add_parser .add_argument ("-n" , dest = " deployment_name" , help = "deployment name" )
512
511
deployment_add_parser .add_argument (
513
- "inference_version" , help = "inference server version (docker image tag)" , default = "latest"
512
+ "-v" , dest = " inference_version" , help = "inference server version (docker image tag)" , default = "latest"
514
513
)
515
514
516
515
deployment_get_parser .set_defaults (func = get_deployment )
517
- deployment_get_parser .add_argument ("api_key" , help = "api key" )
518
- deployment_get_parser .add_argument ("deployment_id" , help = "deployment id" )
516
+ deployment_get_parser .add_argument ("-a" , dest = " api_key" , help = "api key" )
517
+ deployment_get_parser .add_argument ("-d" , dest = " deployment_id" , help = "deployment id" )
519
518
520
519
deployment_list_parser .set_defaults (func = list_deployment )
521
- deployment_list_parser .add_argument ("api_key" , help = "api key" )
520
+ deployment_list_parser .add_argument ("-a" , dest = " api_key" , help = "api key" )
522
521
523
522
deployment_delete_parser .set_defaults (func = delete_deployment )
524
- deployment_delete_parser .add_argument ("api_key" , help = "api key" )
525
- deployment_delete_parser .add_argument ("deployment_id" , help = "deployment id" )
523
+ deployment_delete_parser .add_argument ("-a" , dest = " api_key" , help = "api key" )
524
+ deployment_delete_parser .add_argument ("-d" , dest = " deployment_id" , help = "deployment id" )
526
525
527
526
528
527
def main ():
0 commit comments