I can select rows and columns

src/myfasthtml/controls/DataGrid.py

@@ -179,6 +179,7 @@ class DataGrid(MultipleInstance):
     df = self._df.copy()
     df = self._apply_sort(df)  # need to keep the real type to sort
     df = self._apply_filter(df)
+    self._state.ns_total_rows = len(df)
     
     return df
   
@@ -533,7 +534,8 @@ class DataGrid(MultipleInstance):
     return Div(
       *[Div(selection_type=s_type, element_id=f"{elt_id}") for s_type, elt_id in selected],
       id=f"tsm_{self._id}",
-      selection_mode=f"{self._state.selection.selection_mode}",
+      #selection_mode=f"{self._state.selection.selection_mode}",
+      selection_mode=f"column",
       **extra_attr,
     )