From 26d29f3c02e03b1b51127a642485a25da83e1fe9 Mon Sep 17 00:00:00 2001
From: Hiroki Takizawa <contact@hiroki-takizawa.name>
Date: Wed, 6 Dec 2023 15:54:40 +0900
Subject: [PATCH] fix mypy

---
 python_tests/test_csv_download.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python_tests/test_csv_download.py b/python_tests/test_csv_download.py
index a820f37b..de144fd0 100644
--- a/python_tests/test_csv_download.py
+++ b/python_tests/test_csv_download.py
@@ -1,3 +1,5 @@
+from typing import Any
+
 import optuna
 from optuna_dashboard._app import create_app
 import pytest
@@ -28,7 +30,7 @@ def test_download_csv_fail(id: int) -> None:
 
 @pytest.mark.parametrize("is_multi_obj", [True, False])
 def test_download_csv_multi_obj(is_multi_obj: bool) -> None:
-    def objective(trial: optuna.Trial) -> float:
+    def objective(trial: optuna.Trial) -> Any:
         x = trial.suggest_float("x", -100, 100)
         y = trial.suggest_categorical("y", [-1, 0, 1])
         if is_multi_obj: