From c21d8ef5dc55da16f20c584b3e3ac8142b8e730f Mon Sep 17 00:00:00 2001
From: porink0424 <daichi2120@gmail.com>
Date: Fri, 21 Jun 2024 11:03:33 +0900
Subject: [PATCH] Export useMergedUnionSearchSpace hook from tslib/react

---
 optuna_dashboard/ts/components/GraphContour.tsx            | 7 +++++--
 optuna_dashboard/ts/components/GraphParallelCoordinate.tsx | 7 +++++--
 optuna_dashboard/ts/components/GraphRank.tsx               | 7 +++++--
 optuna_dashboard/ts/components/GraphSlice.tsx              | 7 +++++--
 tslib/react/src/index.ts                                   | 1 +
 5 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/optuna_dashboard/ts/components/GraphContour.tsx b/optuna_dashboard/ts/components/GraphContour.tsx
index 5005883a..76191ce8 100644
--- a/optuna_dashboard/ts/components/GraphContour.tsx
+++ b/optuna_dashboard/ts/components/GraphContour.tsx
@@ -12,14 +12,17 @@ import {
   useTheme,
 } from "@mui/material"
 import blue from "@mui/material/colors/blue"
-import { GraphContainer, useGraphComponentState } from "@optuna/react"
+import {
+  GraphContainer,
+  useGraphComponentState,
+  useMergedUnionSearchSpace,
+} from "@optuna/react"
 import * as plotly from "plotly.js-dist-min"
 import React, { FC, useEffect, useMemo, useState } from "react"
 import { SearchSpaceItem, StudyDetail, Trial } from "ts/types/optuna"
 import { PlotType } from "../apiClient"
 import { getAxisInfo } from "../graphUtil"
 import { usePlot } from "../hooks/usePlot"
-import { useMergedUnionSearchSpace } from "../searchSpace"
 import { usePlotlyColorTheme } from "../state"
 import { useBackendRender } from "../state"
 
diff --git a/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx b/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx
index 22cf3aa9..0f26025d 100644
--- a/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx
+++ b/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx
@@ -6,7 +6,11 @@ import {
   Typography,
   useTheme,
 } from "@mui/material"
-import { GraphContainer, useGraphComponentState } from "@optuna/react"
+import {
+  GraphContainer,
+  useGraphComponentState,
+  useMergedUnionSearchSpace,
+} from "@optuna/react"
 import {
   Target,
   useFilteredTrials,
@@ -19,7 +23,6 @@ import React, { FC, ReactNode, useEffect, useState } from "react"
 import { SearchSpaceItem, StudyDetail } from "ts/types/optuna"
 import { PlotType } from "../apiClient"
 import { usePlot } from "../hooks/usePlot"
-import { useMergedUnionSearchSpace } from "../searchSpace"
 import { usePlotlyColorTheme } from "../state"
 import { useBackendRender } from "../state"
 
diff --git a/optuna_dashboard/ts/components/GraphRank.tsx b/optuna_dashboard/ts/components/GraphRank.tsx
index 5d7e2bdd..bf3c0711 100644
--- a/optuna_dashboard/ts/components/GraphRank.tsx
+++ b/optuna_dashboard/ts/components/GraphRank.tsx
@@ -8,14 +8,17 @@ import {
   Typography,
   useTheme,
 } from "@mui/material"
-import { GraphContainer, useGraphComponentState } from "@optuna/react"
+import {
+  GraphContainer,
+  useGraphComponentState,
+  useMergedUnionSearchSpace,
+} from "@optuna/react"
 import * as plotly from "plotly.js-dist-min"
 import React, { FC, useEffect, useState } from "react"
 import { SearchSpaceItem, StudyDetail, Trial } from "ts/types/optuna"
 import { PlotType } from "../apiClient"
 import { getAxisInfo, makeHovertext } from "../graphUtil"
 import { usePlot } from "../hooks/usePlot"
-import { useMergedUnionSearchSpace } from "../searchSpace"
 import { useBackendRender, usePlotlyColorTheme } from "../state"
 
 const plotDomId = "graph-rank"
diff --git a/optuna_dashboard/ts/components/GraphSlice.tsx b/optuna_dashboard/ts/components/GraphSlice.tsx
index 126b7ff4..d404c2cf 100644
--- a/optuna_dashboard/ts/components/GraphSlice.tsx
+++ b/optuna_dashboard/ts/components/GraphSlice.tsx
@@ -9,7 +9,11 @@ import {
   Typography,
   useTheme,
 } from "@mui/material"
-import { GraphContainer, useGraphComponentState } from "@optuna/react"
+import {
+  GraphContainer,
+  useGraphComponentState,
+  useMergedUnionSearchSpace,
+} from "@optuna/react"
 import {
   Target,
   useFilteredTrials,
@@ -22,7 +26,6 @@ import React, { FC, useEffect, useState } from "react"
 import { SearchSpaceItem, StudyDetail } from "ts/types/optuna"
 import { PlotType } from "../apiClient"
 import { usePlot } from "../hooks/usePlot"
-import { useMergedUnionSearchSpace } from "../searchSpace"
 import { useBackendRender, usePlotlyColorTheme } from "../state"
 
 const plotDomId = "graph-slice"
diff --git a/tslib/react/src/index.ts b/tslib/react/src/index.ts
index 6f236a2e..e1b4243f 100644
--- a/tslib/react/src/index.ts
+++ b/tslib/react/src/index.ts
@@ -17,4 +17,5 @@ export {
   useObjectiveAndUserAttrTargets,
   useObjectiveAndUserAttrTargetsFromStudies,
 } from "./utils/trialFilter"
+export { useMergedUnionSearchSpace } from "./utils/searchSpace"
 export type { GraphComponentState } from "./types"