ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adjust for pandas < 2.0

Browse Source
This commit is contained in:
Alex Burlacu 2023-08-11 19:56:08 +03:00
parent d4b11dfa22
commit 83a04d438c
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
clearml/binding/artifacts.py
View File

@ -1,4 +1,5 @@
import gzip
import io
import json
import yaml
import mimetypes
@ -1062,7 +1063,16 @@ class Artifacts(object):
# (otherwise it is encoded and creates new hash every time)
if self._compression == "gzip":
with gzip.GzipFile(local_filename, 'wb', mtime=0) as gzip_file:
artifact_object.to_csv(gzip_file, **kwargs)
try:
pd_version = int(pd.__version__.split(".")[0])
except ValueError:
pd_version = 0
if pd_version >= 2:
artifact_object.to_csv(gzip_file, **kwargs)
else:
# old (pandas<2) versions of pandas cannot handle direct gzip stream, so we manually encode it
artifact_object.to_csv(io.TextIOWrapper(gzip_file), **kwargs)
else:
artifact_object.to_csv(local_filename, compression=self._compression)