Skip to content

Commit a5b8337

Browse files
committed
fix: update from_dataframe method to handle DataFrameWrapper and convert to view
1 parent 2472853 commit a5b8337

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

python/datafusion/table_provider.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,13 @@ def from_dataframe(cls, df: Any) -> TableProvider:
5454
from datafusion.dataframe import DataFrame as DataFrameWrapper
5555

5656
if isinstance(df, DataFrameWrapper):
57-
df = df.df
57+
dataframe = df
58+
else:
59+
dataframe = DataFrameWrapper(df)
5860

59-
provider = _InternalTableProvider.from_dataframe(df)
61+
provider = dataframe.into_view()
62+
if isinstance(provider, cls):
63+
return provider
6064
return cls(provider)
6165

6266
@classmethod

0 commit comments

Comments
 (0)