Skip to content

Commit 8ee9bfd

Browse files
committed
update to send file
1 parent 3385dce commit 8ee9bfd

File tree

1 file changed

+4
-20
lines changed

1 file changed

+4
-20
lines changed

labelbox/schema/bulk_import_request.py

Lines changed: 4 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -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
6463
def _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

9377
class BulkImportRequest(DbObject):

0 commit comments

Comments
 (0)