ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix dataset CSV files are not previewed properly

Browse Source
This commit is contained in:
allegroai 2023-02-28 17:13:14 +02:00
parent 72b341ee51
commit 487fe75946
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
clearml/datasets/dataset.py
View File

@ -2917,12 +2917,15 @@ class Dataset(object):
if artifact is not None:
# noinspection PyBroadException
try:
# we do not use report_table if default_upload_destination is set because it will
# not upload the sample to that destination, use report_media instead
if isinstance(artifact, pd.DataFrame) and not self._task.get_logger().get_default_upload_destination():
self._task.get_logger().report_table(
"Tables", "summary", table_plot=artifact
)
# we only use report_table if default_upload_destination is not set
# (it is the same as the file server)
# because it will not upload the sample to that destination.
# use report_media instead to not leak data
if (
isinstance(artifact, pd.DataFrame)
and self._task.get_logger().get_default_upload_destination() == Session.get_files_server_host()
):
self._task.get_logger().report_table("Tables", "summary", table_plot=artifact)
else:
self._task.get_logger().report_media(
"Tables", file_name, stream=artifact.to_csv(index=False), file_extension=".txt"