Skip to content

Commit 385cd68

Browse files
committed
change args
1 parent 7e03513 commit 385cd68

File tree

2 files changed

+14
-16
lines changed

2 files changed

+14
-16
lines changed

roboflow/adapters/rfapi.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,14 +207,13 @@ def _save_annotation_error(image_id, response):
207207
return UploadError(errmsg)
208208

209209

210-
def add_deployment(api_key, security_level, cloud_provider, machine_type, deployment_name, inference_version):
210+
def add_deployment(api_key, security_level, machine_type, deployment_name, inference_version):
211211
url = f"{DEDICATED_DEPLOYMENT_URL}/add"
212212
response = requests.post(
213213
url,
214214
json={
215215
"api_key": api_key,
216216
"security_level": security_level,
217-
"cloud_provider": cloud_provider,
218217
"machine_type": machine_type,
219218
"deployment_name": deployment_name,
220219
"inference_version": inference_version,

roboflow/roboflowpy.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,6 @@ def add_deployment(args):
165165
ret_json = rfapi.add_deployment(
166166
args.api_key,
167167
args.security_level,
168-
args.cloud_provider,
169168
args.machine_type,
170169
args.deployment_name,
171170
args.inference_version,
@@ -500,29 +499,29 @@ def _add_deployment_parser(subparsers):
500499
deployment_delete_parser = deployment_subparsers.add_parser("delete", help="delete a dedicated deployment")
501500

502501
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")
506504
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",
510509
)
511-
deployment_add_parser.add_argument("deployment_name", help="deployment name")
510+
deployment_add_parser.add_argument("-n", dest="deployment_name", help="deployment name")
512511
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"
514513
)
515514

516515
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")
519518

520519
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")
522521

523522
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")
526525

527526

528527
def main():

0 commit comments

Comments
 (0)