ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Don't add pandas artifact unnamed:0 column, return original pandas object.

Browse Source
This commit is contained in:
allegroai
2020-09-29 19:21:37 +03:00
parent 30a08ee937
commit 5094ede309
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
trains/binding/artifacts.py
View File

@@ -154,8 +154,10 @@ class Artifact(object):
# noinspection PyProtectedMember
if self.type == 'numpy' and np:
self._object = np.load(local_file)[self.name]
elif self.type in ('pandas', Artifacts._pd_artifact_type) and pd:
elif self.type == Artifacts._pd_artifact_type and pd:
self._object = pd.read_csv(local_file)
elif self.type == 'pandas' and pd:
self._object = pd.read_csv(local_file, index_col=[0])
elif self.type == 'image':
self._object = Image.open(local_file)
elif self.type == 'JSON':