@@ -162,17 +162,27 @@ def infer(args):
162
162
163
163
164
164
def add_deployment (args ):
165
- ret_json = rfapi .add_deployment (args .api_key , args .security_level , args .cloud_provider , args .machine_type , args .deployment_name , args .inference_version )
165
+ ret_json = rfapi .add_deployment (
166
+ args .api_key ,
167
+ args .security_level ,
168
+ args .cloud_provider ,
169
+ args .machine_type ,
170
+ args .deployment_name ,
171
+ args .inference_version ,
172
+ )
166
173
print (json .dumps (ret_json , indent = 2 ))
167
174
175
+
168
176
def get_deployment (args ):
169
177
ret_json = rfapi .get_deployment (args .api_key , args .deployment_id )
170
178
print (json .dumps (ret_json , indent = 2 ))
171
179
180
+
172
181
def list_deployment (args ):
173
182
ret_json = rfapi .list_deployment (args .api_key )
174
183
print (json .dumps (ret_json , indent = 2 ))
175
184
185
+
176
186
def delete_deployment (args ):
177
187
ret_json = rfapi .delete_deployment (args .api_key , args .deployment_id )
178
188
print (json .dumps (ret_json , indent = 2 ))
@@ -483,25 +493,33 @@ def _add_deployment_parser(subparsers):
483
493
)
484
494
deployment_subparsers = deployment_parser .add_subparsers (title = "deployment subcommands" )
485
495
deployment_add_parser = deployment_subparsers .add_parser ("add" , help = "create a dedicated deployment" )
486
- deployment_get_parser = deployment_subparsers .add_parser ("get" , help = "show detailed info for a dedicated deployment" )
496
+ deployment_get_parser = deployment_subparsers .add_parser (
497
+ "get" , help = "show detailed info for a dedicated deployment"
498
+ )
487
499
deployment_list_parser = deployment_subparsers .add_parser ("list" , help = "list dedicated deployments in a workspace" )
488
500
deployment_delete_parser = deployment_subparsers .add_parser ("delete" , help = "delete a dedicated deployment" )
489
501
490
502
deployment_add_parser .set_defaults (func = add_deployment )
491
503
deployment_add_parser .add_argument ("api_key" , help = "api key" )
492
504
deployment_add_parser .add_argument ("security_level" , help = "security level (protected)" , default = "protected" )
493
505
deployment_add_parser .add_argument ("cloud_provider" , help = "cloud provider (gcp)" , default = "gcp" )
494
- deployment_add_parser .add_argument ("machine_type" , help = "machine type (gcp-n1-standard-4-gpu, gcp-g2-standard-8-gpu)" , default = "gcp-n1-standard-4-gpu" )
506
+ 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" ,
510
+ )
495
511
deployment_add_parser .add_argument ("deployment_name" , help = "deployment name" )
496
- deployment_add_parser .add_argument ("inference_version" , help = "inference server version (docker image tag)" , default = "latest" )
497
-
512
+ deployment_add_parser .add_argument (
513
+ "inference_version" , help = "inference server version (docker image tag)" , default = "latest"
514
+ )
515
+
498
516
deployment_get_parser .set_defaults (func = get_deployment )
499
517
deployment_get_parser .add_argument ("api_key" , help = "api key" )
500
518
deployment_get_parser .add_argument ("deployment_id" , help = "deployment id" )
501
-
519
+
502
520
deployment_list_parser .set_defaults (func = list_deployment )
503
521
deployment_list_parser .add_argument ("api_key" , help = "api key" )
504
-
522
+
505
523
deployment_delete_parser .set_defaults (func = delete_deployment )
506
524
deployment_delete_parser .add_argument ("api_key" , help = "api key" )
507
525
deployment_delete_parser .add_argument ("deployment_id" , help = "deployment id" )
0 commit comments