refactor: simplify _repr_html_ method in DataFrame to directly call internal representation

kosiew · kosiew · commit 312fd4ab7e3d · 2025-04-08T13:17:42.000+08:00
diff --git a/python/datafusion/dataframe.py b/python/datafusion/dataframe.py
@@ -152,19 +152,7 @@ def __repr__(self) -> str:
         return self.df.__repr__()
 
     def _repr_html_(self) -> str:
-        """Return HTML representation for Jupyter notebooks."""
-        # Import here to avoid circular imports
-        from datafusion.html_formatter import get_formatter
-
-        # Always get the latest formatter instance
-        formatter = get_formatter()
-
-        # Get data and schema
-        batches = self.collect()
-        schema = self.schema()
-
-        # Format the data using our formatter
-        return formatter.format_html(batches, schema)
+        return self.df._repr_html_()
 
     def describe(self) -> DataFrame:
         """Return the statistics for this DataFrame.
