feat: polars dataframe support

pyproject.toml

@@ -80,6 +80,7 @@ dev = [
     "dask[dataframe]; python_version < '3.7'",
     "playwright; python_version > '3.6'",
     "pytest-playwright; python_version > '3.6'",
+    "polars",
 ]
 assets = [
     "solara-assets==1.22.0"

solara/components/datatable.py

@@ -108,6 +108,8 @@ def DataTable(
 
     if df_type(df) == "pandas":
         column_data = dfs[columns].to_dict("records")
+    elif df_type(df) == "polars":
+        column_data = dfs[columns].to_dicts()
     else:
         column_data = dfs[columns].to_records()
     for i in range(i2 - i1):

solara/lab/hooks/dataframe.py

@@ -6,5 +6,7 @@ def use_df_column_names(df):
         return df.get_column_names()
     elif df_type(df) == "pandas":
         return df.columns.tolist()
+    elif df_type(df) == "polars":
+        return df.columns
     else:
         raise TypeError(f"{type(df)} not supported")

tests/unit/datatable_test.py

@@ -1,13 +1,15 @@
 import pytest
 import solara
 import vaex
+import polars as pl
 from solara.components.datatable import DataTable, DataTableWidget
 
 df_vaex = vaex.datasets.titanic()
 df_pandas = df_vaex.to_pandas_df()
+df_polars = pl.from_pandas(df_pandas)
 
 
-@pytest.mark.parametrize("df", [df_vaex, df_pandas])
+@pytest.mark.parametrize("df", [df_vaex, df_pandas, df_polars])
 def test_render(df):
     @solara.component
     def Test():