From ab7257e9b025d510020732bff02e30469cab4eac Mon Sep 17 00:00:00 2001
From: keisuke-umezawa <keisuke.umezawa@gmail.com>
Date: Mon, 15 Apr 2024 14:59:20 +0900
Subject: [PATCH] Changed default row numbers

---
 optuna_dashboard/ts/components/DataGrid.tsx   | 7 +++++++
 optuna_dashboard/ts/components/TrialTable.tsx | 4 ++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/optuna_dashboard/ts/components/DataGrid.tsx b/optuna_dashboard/ts/components/DataGrid.tsx
index 7f858ee8..4efecfd4 100644
--- a/optuna_dashboard/ts/components/DataGrid.tsx
+++ b/optuna_dashboard/ts/components/DataGrid.tsx
@@ -45,6 +45,7 @@ import {
   getFacetedUniqueValues,
   getFacetedRowModel,
   SortingState,
+  PaginationState,
   ColumnFiltersState,
   useReactTable,
 } from "@tanstack/react-table"
@@ -492,6 +493,10 @@ function DataGrid2(props: {
   )
   const [filterMenuAnchorEl, setFilterMenuAnchorEl] =
     React.useState<null | HTMLElement>(null)
+  const [pagination, setPagination] = React.useState<PaginationState>({
+    pageIndex: 0,
+    pageSize: 50,
+  })
 
   const table = useReactTable({
     data,
@@ -499,9 +504,11 @@ function DataGrid2(props: {
     state: {
       columnFilters,
       sorting,
+      pagination,
     },
     onColumnFiltersChange: setColumnFilters,
     onSortingChange: setSorting,
+    onPaginationChange: setPagination,
     getCoreRowModel: getCoreRowModel(),
     getFacetedRowModel: getFacetedRowModel(),
     getFacetedUniqueValues: getFacetedUniqueValues(),
diff --git a/optuna_dashboard/ts/components/TrialTable.tsx b/optuna_dashboard/ts/components/TrialTable.tsx
index 7d78e967..7ea88c6f 100644
--- a/optuna_dashboard/ts/components/TrialTable.tsx
+++ b/optuna_dashboard/ts/components/TrialTable.tsx
@@ -19,7 +19,7 @@ import {
   FilterFn,
 } from "@tanstack/react-table"
 
-const multiValueFilter: FilterFn<any> = <D extends object>(
+const multiValueFilter: FilterFn<Trial> = <D extends object>(
   row: Row<D>,
   columnId: IdType<D>,
   filterValue: string[]
@@ -49,7 +49,7 @@ export const TrialTable: FC<{
   ]
 
   const columnHelper = createColumnHelper<Trial>()
-  const tcolumns: ColumnDef<Trial, any>[] = [
+  const tcolumns: ColumnDef<Trial>[] = [
     columnHelper.accessor("number", {
       header: "Number",
       footer: (info) => info.column.id,