Skip to content

Commit 7e03513

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent 66d9a0f commit 7e03513

File tree

3 files changed

+43
-15
lines changed

3 files changed

+43
-15
lines changed

roboflow/adapters/rfapi.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import requests
77
from requests_toolbelt.multipart.encoder import MultipartEncoder
88

9-
from roboflow.config import API_URL, DEFAULT_BATCH_NAME, DEFAULT_JOB_NAME, DEDICATED_DEPLOYMENT_URL
9+
from roboflow.config import API_URL, DEDICATED_DEPLOYMENT_URL, DEFAULT_BATCH_NAME, DEFAULT_JOB_NAME
1010
from roboflow.util import image_utils
1111

1212

@@ -209,9 +209,17 @@ def _save_annotation_error(image_id, response):
209209

210210
def add_deployment(api_key, security_level, cloud_provider, machine_type, deployment_name, inference_version):
211211
url = f"{DEDICATED_DEPLOYMENT_URL}/add"
212-
response = requests.post(url, json={'api_key': api_key, 'security_level': security_level,
213-
'cloud_provider': cloud_provider, 'machine_type': machine_type,
214-
'deployment_name': deployment_name, 'inference_version': inference_version})
212+
response = requests.post(
213+
url,
214+
json={
215+
"api_key": api_key,
216+
"security_level": security_level,
217+
"cloud_provider": cloud_provider,
218+
"machine_type": machine_type,
219+
"deployment_name": deployment_name,
220+
"inference_version": inference_version,
221+
},
222+
)
215223
if response.status_code != 200:
216224
raise RoboflowError(response.text)
217225
result = response.json()
@@ -220,7 +228,7 @@ def add_deployment(api_key, security_level, cloud_provider, machine_type, deploy
220228

221229
def get_deployment(api_key, deployment_id):
222230
url = f"{DEDICATED_DEPLOYMENT_URL}/get"
223-
response = requests.get(url, json={'api_key': api_key, 'deployment_id': deployment_id})
231+
response = requests.get(url, json={"api_key": api_key, "deployment_id": deployment_id})
224232
if response.status_code != 200:
225233
raise RoboflowError(response.text)
226234
result = response.json()
@@ -229,7 +237,7 @@ def get_deployment(api_key, deployment_id):
229237

230238
def list_deployment(api_key):
231239
url = f"{DEDICATED_DEPLOYMENT_URL}/list"
232-
response = requests.get(url, json={'api_key': api_key})
240+
response = requests.get(url, json={"api_key": api_key})
233241
if response.status_code != 200:
234242
raise RoboflowError(response.text)
235243
result = response.json()
@@ -238,7 +246,7 @@ def list_deployment(api_key):
238246

239247
def delete_deployment(api_key, deployment_id):
240248
url = f"{DEDICATED_DEPLOYMENT_URL}/delete"
241-
response = requests.post(url, json={'api_key': api_key, 'deployment_id': deployment_id})
249+
response = requests.post(url, json={"api_key": api_key, "deployment_id": deployment_id})
242250
if response.status_code != 200:
243251
raise RoboflowError(response.text)
244252
result = response.json()

roboflow/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,9 @@ def get_conditional_configuration_variable(key, default):
6363
CLIP_FEATURIZE_URL = get_conditional_configuration_variable("CLIP_FEATURIZE_URL", "CLIP FEATURIZE URL NOT IN ENV")
6464
OCR_URL = get_conditional_configuration_variable("OCR_URL", "OCR URL NOT IN ENV")
6565

66-
DEDICATED_DEPLOYMENT_URL = get_conditional_configuration_variable("DEDICATED_DEPLOYMENT_URL", "https://deployment.svc.roboflow.one")
66+
DEDICATED_DEPLOYMENT_URL = get_conditional_configuration_variable(
67+
"DEDICATED_DEPLOYMENT_URL", "https://deployment.svc.roboflow.one"
68+
)
6769

6870
DEMO_KEYS = ["coco-128-sample", "chess-sample-only-api-key"]
6971

roboflow/roboflowpy.py

Lines changed: 25 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -162,17 +162,27 @@ def infer(args):
162162

163163

164164
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+
)
166173
print(json.dumps(ret_json, indent=2))
167174

175+
168176
def get_deployment(args):
169177
ret_json = rfapi.get_deployment(args.api_key, args.deployment_id)
170178
print(json.dumps(ret_json, indent=2))
171179

180+
172181
def list_deployment(args):
173182
ret_json = rfapi.list_deployment(args.api_key)
174183
print(json.dumps(ret_json, indent=2))
175184

185+
176186
def delete_deployment(args):
177187
ret_json = rfapi.delete_deployment(args.api_key, args.deployment_id)
178188
print(json.dumps(ret_json, indent=2))
@@ -483,25 +493,33 @@ def _add_deployment_parser(subparsers):
483493
)
484494
deployment_subparsers = deployment_parser.add_subparsers(title="deployment subcommands")
485495
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+
)
487499
deployment_list_parser = deployment_subparsers.add_parser("list", help="list dedicated deployments in a workspace")
488500
deployment_delete_parser = deployment_subparsers.add_parser("delete", help="delete a dedicated deployment")
489501

490502
deployment_add_parser.set_defaults(func=add_deployment)
491503
deployment_add_parser.add_argument("api_key", help="api key")
492504
deployment_add_parser.add_argument("security_level", help="security level (protected)", default="protected")
493505
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+
)
495511
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+
498516
deployment_get_parser.set_defaults(func=get_deployment)
499517
deployment_get_parser.add_argument("api_key", help="api key")
500518
deployment_get_parser.add_argument("deployment_id", help="deployment id")
501-
519+
502520
deployment_list_parser.set_defaults(func=list_deployment)
503521
deployment_list_parser.add_argument("api_key", help="api key")
504-
522+
505523
deployment_delete_parser.set_defaults(func=delete_deployment)
506524
deployment_delete_parser.add_argument("api_key", help="api key")
507525
deployment_delete_parser.add_argument("deployment_id", help="deployment id")

0 commit comments

Comments
 (0)