Skip to content

Commit 67d89c7

Browse files
committed
fix: smartdatalake to convert on pd.Dataframe to pai.Dataframe
1 parent 4cd1de1 commit 67d89c7

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pandasai/smart_datalake/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,11 @@ def load_dfs(self, dfs: List[pd.DataFrame]):
3434
load_dfs = []
3535
for df in dfs:
3636
if isinstance(df, pd.DataFrame):
37-
load_dfs.append(DataFrame(df))
37+
load_dfs.append(
38+
DataFrame(df)
39+
if not isinstance(df, DataFrame) and isinstance(df, pd.DataFrame)
40+
else df
41+
)
3842
else:
3943
raise ValueError(
4044
"Invalid input data. We cannot convert it to a dataframe."

0 commit comments

Comments
 (0)