Skip to content

Commit 3eb1e43

Browse files
committed
linting
1 parent d345a9b commit 3eb1e43

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

roboflow/models/inference.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -260,8 +260,8 @@ def predict_video(
260260
models = [
261261
{
262262
"model_id": self.dataset_id,
263-
"model_version":self.version,
264-
"inference_type": self.type
263+
"model_version": self.version,
264+
"inference_type": self.type,
265265
}
266266
]
267267

@@ -324,12 +324,12 @@ def poll_for_video_results(self, job_id: str = None) -> dict:
324324
raise Exception(f"Error getting video inference results: {response.text}")
325325
data = response.json()
326326
if "status" not in data:
327-
return {} # No status available
327+
return {} # No status available
328328
if data.get("status") > 1:
329-
return(data) # Error
329+
return data # Error
330330
elif data.get("status") == 1:
331331
return {} # Still running
332-
else: # done
332+
else: # done
333333
output_signed_url = data["output_signed_url"]
334334
inference_data = requests.get(
335335
output_signed_url, headers={"Content-Type": "application/json"}

0 commit comments

Comments
 (0)