From a2cf3ebd54d3dbe2b23cd2bb4fbfc554f9b3cd41 Mon Sep 17 00:00:00 2001 From: nabenabe0928 Date: Tue, 26 Dec 2023 09:15:55 +0100 Subject: [PATCH] Apply formatter --- python_tests/test_api.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/python_tests/test_api.py b/python_tests/test_api.py index ebe5a0ef..cbf36e0c 100644 --- a/python_tests/test_api.py +++ b/python_tests/test_api.py @@ -1,14 +1,13 @@ from __future__ import annotations import json -import pytest import sys +from typing import Any from unittest import TestCase import optuna from optuna import get_all_study_summaries from optuna.study import StudyDirection -from optuna.trial import FrozenTrial from optuna_dashboard._app import create_app from optuna_dashboard._app import create_new_study from optuna_dashboard._note import note_str_key_prefix @@ -47,8 +46,10 @@ class APITestCase(TestCase): self.assertEqual(len(study_summaries), 2) def run_get_study_details( - self, queries: dict[str, str] | None = None, expected_status: int = 200, - ) -> tuple[int, list[FrozenTrial]]: + self, + queries: dict[str, str] | None = None, + expected_status: int = 200, + ) -> list[dict[str, Any]]: study = optuna.create_study() study_id = study._study_id study.optimize(objective, n_trials=10)