File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff 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" }
You can’t perform that action at this time.
0 commit comments