wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-02 12:41:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add # type: ignore to python_tests/test_api.py.

Browse Source
This commit is contained in:
gen740
2022-12-01 15:11:26 +09:00
parent abefb6d030
commit 83e8aec664
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_api.py
+15 -5
View File
@@ -21,8 +21,12 @@ class APITestCase(TestCase):
storage = optuna.storages.InMemoryStorage()
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(
study_name="foo1", directions=[StudyDirection.MINIMIZE]
) # type: ignore
storage.create_new_study(
study_name="foo2", directions=[StudyDirection.MINIMIZE]
) # type: ignore
else:
storage.create_new_study("foo1")
storage.create_new_study("foo2")
@@ -135,7 +139,9 @@ class APITestCase(TestCase):
def test_create_study_duplicated(self) -> None:
storage = optuna.storages.InMemoryStorage()
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(
study_name="foo", directions=[StudyDirection.MINIMIZE]
) # type: ignore
else:
storage.create_new_study("foo")
self.assertEqual(len(get_all_study_summaries(storage)), 1)
@@ -158,8 +164,12 @@ class APITestCase(TestCase):
def test_delete_study(self) -> None:
storage = optuna.storages.InMemoryStorage()
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(
study_name="foo1", directions=[StudyDirection.MINIMIZE]
) # type: ignore
storage.create_new_study(
study_name="foo2", directions=[StudyDirection.MINIMIZE]
) # type: ignore
else:
storage.create_new_study(study_name="foo1")
storage.create_new_study(study_name="foo2")