@@ -60,34 +60,18 @@ def _make_request_data(project_id: str, name: str, content_length: int,
60
60
}
61
61
62
62
63
- # TODO(gszpak): move it to client.py
64
63
def _send_create_file_command (
65
64
client , request_data : dict , file_name : str ,
66
65
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 )
78
66
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 })
83
68
84
- if not response_data .get ("createBulkImportRequest" , None ):
69
+ if not response .get ("createBulkImportRequest" , None ):
85
70
raise labelbox .exceptions .LabelboxError (
86
71
"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 ))
89
73
90
- return response_data
74
+ return response
91
75
92
76
93
77
class BulkImportRequest (DbObject ):
0 commit comments