Skip to content

Commit 3f9bd40

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent db0c449 commit 3f9bd40

File tree

2 files changed

+35
-14
lines changed

2 files changed

+35
-14
lines changed

roboflow/adapters/rfapi.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -209,18 +209,26 @@ def _save_annotation_error(image_id, response):
209209

210210
def add_runner(api_key, security_level, cloud_provider, machine_type, runner_name, inference_version):
211211
url = f"{ROBOFLOW_RUNNER_URL}/create"
212-
response = requests.post(url, json={'api_key': api_key, 'security_level': security_level,
213-
'cloud_provider': cloud_provider, 'machine_type': machine_type,
214-
'runner_name': runner_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+
"runner_name": runner_name,
220+
"inference_version": inference_version,
221+
},
222+
)
215223
if response.status_code != 200:
216224
raise RoboflowError(response.text)
217225
result = response.json()
218226
return result
219-
227+
220228

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

230238
def list_runner(api_key):
231239
url = f"{ROBOFLOW_RUNNER_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,9 +246,8 @@ def list_runner(api_key):
238246

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

roboflow/roboflowpy.py

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

163163

164164
def add_runner(args):
165-
runner_json = rfapi.add_runner(args.api_key, args.security_level, args.cloud_provider, args.machine_type, args.runner_name, args.inference_version)
165+
runner_json = rfapi.add_runner(
166+
args.api_key,
167+
args.security_level,
168+
args.cloud_provider,
169+
args.machine_type,
170+
args.runner_name,
171+
args.inference_version,
172+
)
166173
print(json.dumps(runner_json, indent=2))
167174

175+
168176
def get_runner(args):
169177
runner_json = rfapi.get_runner(args.api_key, args.runner_id)
170178
print(json.dumps(runner_json, indent=2))
171179

180+
172181
def list_runner(args):
173182
runner_json = rfapi.list_runner(args.vapi_key)
174183
print(json.dumps(runner_json, indent=2))
175184

185+
176186
def delete_runner(args):
177187
runner_json = rfapi.delete_runner(args.api_key, args.runner_id)
178188
print(json.dumps(runner_json, indent=2))
@@ -491,17 +501,21 @@ def _add_runner_parser(subparsers):
491501
runner_add_parser.add_argument("api_key", help="api key")
492502
runner_add_parser.add_argument("security_level", help="security level (protected)", default="protected")
493503
runner_add_parser.add_argument("cloud_provider", help="cloud provider (gcp)", default="gcp")
494-
runner_add_parser.add_argument("machine_type", help="machine type (n1-standard-4-cpu, g2-standard-8-cpu)", default="n1-standard-4-cpu")
504+
runner_add_parser.add_argument(
505+
"machine_type", help="machine type (n1-standard-4-cpu, g2-standard-8-cpu)", default="n1-standard-4-cpu"
506+
)
495507
runner_add_parser.add_argument("runner_name", help="runner name")
496-
runner_add_parser.add_argument("inference_version", help="inference server version (docker image tag)", default="latest")
497-
508+
runner_add_parser.add_argument(
509+
"inference_version", help="inference server version (docker image tag)", default="latest"
510+
)
511+
498512
runner_get_parser.set_defaults(func=get_runner)
499513
runner_get_parser.add_argument("api_key", help="api key")
500514
runner_get_parser.add_argument("runner_id", help="runner id")
501-
515+
502516
runner_list_parser.set_defaults(func=list_runner)
503517
runner_list_parser.add_argument("api_key", help="api key")
504-
518+
505519
runner_delete_parser.set_defaults(func=delete_runner)
506520
runner_delete_parser.add_argument("api_key", help="api key")
507521
runner_delete_parser.add_argument("runner_id", help="runner id")

0 commit comments

Comments
 (0)