From 5bab40e33c357e0ccb344b1ff40af5aec8aa56da Mon Sep 17 00:00:00 2001 From: keisuke-umezawa Date: Sat, 2 Mar 2024 22:13:44 +0900 Subject: [PATCH] Fix typo --- e2e_tests/test_standalone/test_study_list.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/e2e_tests/test_standalone/test_study_list.py b/e2e_tests/test_standalone/test_study_list.py index ed749979..9ac9f221 100644 --- a/e2e_tests/test_standalone/test_study_list.py +++ b/e2e_tests/test_standalone/test_study_list.py @@ -28,7 +28,7 @@ def test_home( assert title == "Optuna Dashboard (Wasm ver.)" -def create_rdb_storage_file(filename: str, study_name: str, storage_type: str): +def create_storage_file(filename: str, study_name: str, storage_type: str): if storage_type == "rdb": storage = optuna.storages.RDBStorage(f"sqlite:///{filename}") elif storage_type == "journal": @@ -52,7 +52,7 @@ def create_rdb_storage_file(filename: str, study_name: str, storage_type: str): def test_load_storage( page: Page, server_url: str, - create_storage_file: Callable[[str, str], optuna.storages.BaseStorage], + storage_type: str, ) -> None: study_name = "single-objective" url = f"{server_url}" @@ -61,7 +61,7 @@ def test_load_storage( with tempfile.NamedTemporaryFile() as fp: filename = fp.name path = os.path.join(dir, filename) - create_storage_file(filename, study_name) + create_storage_file(filename, study_name, storage_type) page.goto(url) with page.expect_file_chooser() as fc_info: page.get_by_role("button").nth(2).click()