@@ -60,34 +60,18 @@ def _make_request_data(project_id: str, name: str, content_length: int,
6060 }
6161
6262
63- # TODO(gszpak): move it to client.py
6463def _send_create_file_command (
6564 client , request_data : dict , file_name : str ,
6665 file_data : Tuple [str , Union [bytes , BinaryIO ], str ]) -> dict :
67- response = requests .post (
68- client .endpoint ,
69- headers = {"authorization" : "Bearer %s" % client .api_key },
70- data = request_data ,
71- files = {file_name : file_data })
72-
73- try :
74- response_json = response .json ()
75- except ValueError :
76- raise labelbox .exceptions .LabelboxError (
77- "Failed to parse response as JSON: %s" % response .text )
7866
79- response_data = response_json .get ("data" , None )
80- if response_data is None :
81- raise labelbox .exceptions .LabelboxError (
82- "Failed to upload, message: %s" % response_json .get ("errors" , None ))
67+ response = client .execute (data = request_data , files = {file_name : file_data })
8368
84- if not response_data .get ("createBulkImportRequest" , None ):
69+ if not response .get ("createBulkImportRequest" , None ):
8570 raise labelbox .exceptions .LabelboxError (
8671 "Failed to create BulkImportRequest, message: %s" %
87- response_json .get ("errors" , None ) or
88- response_data .get ("error" , None ))
72+ response .get ("errors" , None ) or response .get ("error" , None ))
8973
90- return response_data
74+ return response
9175
9276
9377class BulkImportRequest (DbObject ):
0 commit comments