ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +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
Show Stats 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':

4
trains/logger.py
View File

@ -309,9 +309,9 @@ class Logger(object):
"please install the pandas python package"
)
if url:
table = pd.read_csv(url)
table = pd.read_csv(url, index_col=[0])
elif csv:
table = pd.read_csv(csv)
table = pd.read_csv(csv, index_col=[0])
def replace(dst, *srcs):
for src in srcs: