<Button size="small" onClick={toggle2ndGroup}>
  Toggle 2nd row expansion
</Button>
<Box sx={{ height: 400, pt: 1 }}>
  <DataGridPremium
    {...data}
    apiRef={apiRef}
    disableRowSelectionOnClick
    initialState={initialState}
  />
</Box>