ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +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,
num_download_attempts=container.config.retries),
Callback=callback,
ExtraArgs={'ContentType': get_file_mimetype(file_path)}
ExtraArgs={'ContentType': get_file_mimetype(object_name or file_path)}
)
except Exception as ex:
self.get_logger().error('Failed uploading: %s' % ex)
@ -1970,7 +1970,7 @@ class _AzureBlobServiceStorageDriver(_Driver):
blob_name,
file_path,
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,
)
return True
@ -2644,7 +2644,7 @@ def get_file_mimetype(file_path):
# noinspection PyBroadException
try:
file_path = Path(file_path).resolve()
mimetype, _ = mimetypes.guess_type(file_path)
mimetype, _ = mimetypes.guess_type(file_path.as_posix())
if mimetype:
return mimetype
except Exception: