Skip to content

Commit bda26be

Browse files
committed
Revert "Removing get_sample_df to remove pandas, considered a too heavy addition." and importing pandas locally.
1 parent 4d9bf43 commit bda26be

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

dataikuapi/dss/modelevaluationstore.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
import json
22
from io import BytesIO
33

4-
import pandas as pd
5-
64
from dataikuapi.dss.metrics import ComputedMetrics
75
from .discussion import DSSObjectDiscussions
86

@@ -315,6 +313,7 @@ def get_sample_df(self):
315313
"/projects/%s/modelevaluationstores/%s/runs/%s/schema" % (self.project_key, self.mes_id, self.run_id)
316314
).text
317315
schema = json.loads(schema_txt)
316+
import pandas as pd
318317
return pd.read_csv(BytesIO(buf.getvalue()), compression='gzip', sep='\t', header=None, names=[c["name"] for c in schema["columns"]])
319318

320319

0 commit comments

Comments
 (0)