Fixed type hint and format function call

widgetti · Apr 29, 2024 · 5e75e74 · 5e75e74
1 parent 1c1f29f
commit 5e75e74
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/solara/components/datatable.py b/solara/components/datatable.py
@@ -105,7 +105,7 @@ def DataTable(
     dfs = df_slice(df, i1, i2)
     records = df_records(dfs)
     for i in range(i2 - i1):
-        item = {"__row__": format(rows[i + i1])}  # special key for the row number
+        item = {"__row__": format(dfs, columns, i + 1, rows[i + i1])}  # special key for the row number
         for column in columns:
             item[column] = format(dfs, column, i + i1, records[i][column])
         items.append(item)

diff --git a/solara/lab/utils/dataframe.py b/solara/lab/utils/dataframe.py
@@ -27,7 +27,7 @@ def df_columns(df) -> List[str]:
     else:
         raise TypeError(f"{type(df)} not supported")
 
-def df_row_names(df) -> List[any]:
+def df_row_names(df) -> List[Union[int, str]]:
     """Return a list of row names from a dataframe."""
     if df_type(df) == "vaex" or df_type(df) == "polars":
         return list(range(df_len(df)))