ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-05-02 12:02:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix upload to S3 guess mimetypes

Browse Source
This commit is contained in:
allegroai 2022-07-21 17:18:53 +03:00
parent a1edcea481
commit beb1bd447b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/storage/helper.py
View File

@ -1454,7 +1454,7 @@ class _Boto3Driver(_Driver):
max_concurrency=self._max_multipart_concurrency if container.config.multipart else 1, max_concurrency=self._max_multipart_concurrency if container.config.multipart else 1,
num_download_attempts=container.config.retries), num_download_attempts=container.config.retries),
Callback=callback, Callback=callback,
ExtraArgs={'ContentType': get_file_mimetype(file_path)} ExtraArgs={'ContentType': get_file_mimetype(object_name or file_path)}
) )
except Exception as ex: except Exception as ex:
self.get_logger().error('Failed uploading: %s' % ex) self.get_logger().error('Failed uploading: %s' % ex)
@ -1970,7 +1970,7 @@ class _AzureBlobServiceStorageDriver(_Driver):
blob_name, blob_name,
file_path, file_path,
max_connections=2, max_connections=2,
content_settings=ContentSettings(content_type=get_file_mimetype(file_path)), content_settings=ContentSettings(content_type=get_file_mimetype(object_name or file_path)),
progress_callback=callback, progress_callback=callback,
) )
return True return True
@ -2644,7 +2644,7 @@ def get_file_mimetype(file_path):
# noinspection PyBroadException # noinspection PyBroadException
try: try:
file_path = Path(file_path).resolve() file_path = Path(file_path).resolve()
mimetype, _ = mimetypes.guess_type(file_path) mimetype, _ = mimetypes.guess_type(file_path.as_posix())
if mimetype: if mimetype:
return mimetype return mimetype
except Exception: except Exception: