DataGrid not showing using ui.output_data_frame

I'm using DataGrid and render.data_frame to be able to do row selection but the data frame is not displaying. When debugging, the returned object is a DataGrid shiny object. If I'm using basic pd.DataFrame everything is right.

Here, the server part :

@output
@render.data_frame
def feature_selection():
  meth = input.meth_graph()
  results = tranformed_data().dict_results[meth][2]
  to_return =  pd.DataFrame([raw_data().features,results]).T
  return render.DataGrid(
    to_return,
    row_selection_mode = 'multiple')

Here the ui part :

ui.nav(
  ui.row(
    ui.output_data_frame('feature_selection')
  )
)

Start by reproducing the provided server and UI code with render.DataGrid and ui.output_data_frame, then inspect the data-frame rendering path and its handling of DataGrid objects. Done means the returned DataGrid is displayed and multiple row selection works as intended.