//Im facing the same issue 
    //Now I fixed with using 


    //sx={{display: 'grid'}}

      <DataGrid
          rows={rows}
          sx={{display: 'grid'}}
          columns={columns}
          checkboxSelection
          onRowSelectionModelChange={(selectionModel) =>
            handleRowSelectionChange(selectionModel as GridRowId[]) 
            
          }
          slots={{
            toolbar: () => (
              <CustomToolbar
                exportToPDF={exportToPDF}
              />
            ),
            noRowsOverlay: NoRowsOverlay,
          }}
          disableRowSelectionOnClick
          density='compact'
          pagination
          paginationModel={paginationModel}
          rowCount={rowCount}
          paginationMode={paginationMode}
          pageSizeOptions={[5, 10, 20, 50]}
          onPaginationModelChange={handlePaginationModelChange}
        />