Skip to content

Commit f6fe970

Browse files
committed
refactor: simplify from_dataframe method to directly return DataFrame view
1 parent c99f7f7 commit f6fe970

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

python/datafusion/table_provider.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -56,14 +56,8 @@ def from_dataframe(cls, df: Any) -> TableProvider:
5656
"""Create a :class:`TableProvider` from a :class:`DataFrame`."""
5757
from datafusion.dataframe import DataFrame as DataFrameWrapper
5858

59-
dataframe = (
60-
df if isinstance(df, DataFrameWrapper) else DataFrameWrapper(df)
61-
)
62-
63-
provider = dataframe.into_view()
64-
if isinstance(provider, cls):
65-
return provider
66-
return cls(provider)
59+
df = df if isinstance(df, DataFrameWrapper) else DataFrameWrapper(df)
60+
return df.into_view()
6761

6862
@classmethod
6963
def from_view(cls, df: Any) -> TableProvider:

0 commit comments

Comments
 (0)