fix by lint

This commit is contained in:
moririn2528
2023-08-30 14:36:58 +09:00
parent 294c47f338
commit 9d1c006f9e
5 changed files with 10 additions and 17 deletions
-1
View File
@@ -140,7 +140,6 @@ class APITestCase(TestCase):
{
"candidate_trials": [0, 1, 2],
"preferentials": [[0, 1], [2, 1]],
"mode": "choose_worst",
}
),
content_type="application/json",
+3 -8
View File
@@ -3,13 +3,12 @@ from __future__ import annotations
from datetime import datetime
from typing import Callable
from optuna_dashboard._preferential_history import report_choice
from optuna_dashboard._serializer import serialize_preference_history
from optuna_dashboard.preferential import create_study
from optuna_dashboard.preferential._history import FeedbackMode
from optuna_dashboard.preferential._history import report_choice
from ..storage_supplier import parametrize_storages
from ..storage_supplier import StorageSupplier
from .storage_supplier import parametrize_storages
from .storage_supplier import StorageSupplier
@parametrize_storages
@@ -27,7 +26,6 @@ def test_report_and_get_choices(storage_supplier: Callable[[], StorageSupplier])
storage=storage,
candidate_trials=[0, 2, 3, 4],
preferences=[(2, 0), (3, 0), (4, 0)],
feedback_mode=FeedbackMode.CHOOSE_WORST,
timestamp=datetime(2020, 1, 1, 10, 0, 1),
)
report_choice(
@@ -35,7 +33,6 @@ def test_report_and_get_choices(storage_supplier: Callable[[], StorageSupplier])
storage=storage,
candidate_trials=[0, 1, 2],
preferences=[(0, 1), (2, 1)],
feedback_mode=FeedbackMode.CHOOSE_WORST,
timestamp=datetime(2020, 1, 1, 10, 0, 0),
)
@@ -47,7 +44,6 @@ def test_report_and_get_choices(storage_supplier: Callable[[], StorageSupplier])
assert len(history[0]["preferences"][i]) == 2
assert history[0]["preferences"][i][0] == best
assert history[0]["preferences"][i][1] == worst
assert history[0]["feedback_mode"] == FeedbackMode.CHOOSE_WORST.name
assert history[0]["timestamp"] == "2020-01-01T10:00:00"
assert history[1]["candidate_trials"] == [0, 2, 3, 4]
assert len(history[1]["preferences"]) == 3
@@ -55,5 +51,4 @@ def test_report_and_get_choices(storage_supplier: Callable[[], StorageSupplier])
assert len(history[1]["preferences"][i]) == 2
assert history[1]["preferences"][i][0] == best
assert history[1]["preferences"][i][1] == worst
assert history[1]["feedback_mode"] == FeedbackMode.CHOOSE_WORST.name
assert history[1]["timestamp"] == "2020-01-01T10:00:01"