ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-16 21:42:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add polars support to Model

Browse Source
This commit is contained in:
BlakeJC94 2024-12-02 22:08:05 +11:00
parent 3c38f008a0
commit 969a48906b
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
clearml/model.py
View File

@ -14,6 +14,10 @@ try:
import pandas as pd import pandas as pd
except ImportError: except ImportError:
pd = None pd = None
try:
import polars as pl
except ImportError:
pl = None
from .backend_api import Session from .backend_api import Session
from .backend_api.services import models, projects from .backend_api.services import models, projects
@ -638,15 +642,21 @@ class BaseModel(object):
) )
table = table_plot table = table_plot
if url or csv: if url or csv:
if not pd: if not pd and not pl:
raise UsageError( raise UsageError(
"pandas is required in order to support reporting tables using CSV or a URL, " "pandas or polars is required in order to support reporting tables using CSV or a URL, "
"please install the pandas python package" "please install the pandas or polars python package"
) )
if url: if url:
table = pd.read_csv(url, index_col=[0]) if pd:
table = pd.read_csv(url, index_col=[0])
else:
table = pd.read_csv(url)
elif csv: elif csv:
table = pd.read_csv(csv, index_col=[0]) if pd:
table = pd.read_csv(csv, index_col=[0])
else:
table = pd.read_csv(url)
def replace(dst, *srcs): def replace(dst, *srcs):
for src in srcs: for src in srcs: