diff --git a/visual_regression_test.py b/visual_regression_test.py index c0c2efa4..be7463c7 100644 --- a/visual_regression_test.py +++ b/visual_regression_test.py @@ -164,6 +164,7 @@ async def take_screenshots(storage: optuna.storages.BaseStorage) -> None: def main() -> None: os.makedirs(args.output_dir, exist_ok=True) + storage: optuna.storages.BaseStorage if args.storage: storage = optuna.storages.RDBStorage(args.storage) else: