diff --git a/e2e_tests/test_dashboard/visual_regression_test.py b/e2e_tests/test_dashboard/visual_regression_test.py index ea6ff930..876c585b 100644 --- a/e2e_tests/test_dashboard/visual_regression_test.py +++ b/e2e_tests/test_dashboard/visual_regression_test.py @@ -4,11 +4,22 @@ import optuna from playwright.sync_api import Page import pytest +from optuna_dashboard._storage import trials_cache +from optuna_dashboard._storage import trials_cache_lock +from optuna_dashboard._storage import trials_last_fetched_at + from ..test_server import make_test_server +def clear_inmemory_cache() -> None: + with trials_cache_lock: + trials_cache.clear() + trials_last_fetched_at.clear() + + @pytest.fixture def storage() -> optuna.storages.InMemoryStorage: + clear_inmemory_cache() storage = optuna.storages.InMemoryStorage() return storage