Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions src/frontend_visualqa/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from collections.abc import AsyncIterator, Callable, Coroutine
from contextlib import asynccontextmanager
from pathlib import Path
from typing import Any, TypeVar
from typing import TYPE_CHECKING, Any, TypeVar

from pydantic import ValidationError

Expand All @@ -33,6 +33,9 @@
)
from frontend_visualqa.text_utils import clip_text

if TYPE_CHECKING:
from frontend_visualqa.runner import VisualQARunner

logger = logging.getLogger(__name__)

T = TypeVar("T")
Expand Down Expand Up @@ -601,7 +604,7 @@ async def _run_status() -> dict[str, Any]:
return serialize_result(result)


def _new_runner(*, browser_config: BrowserConfig | None = None, reporters: list[str] | None = None) -> Any:
def _new_runner(*, browser_config: BrowserConfig | None = None, reporters: list[str] | None = None) -> VisualQARunner:
from frontend_visualqa.runner import VisualQARunner

return VisualQARunner(browser_config=browser_config, reporters=reporters)
Expand All @@ -610,7 +613,7 @@ def _new_runner(*, browser_config: BrowserConfig | None = None, reporters: list[
@asynccontextmanager
async def _runner_scope(
*, browser_config: BrowserConfig | None = None, reporters: list[str] | None = None
) -> AsyncIterator[Any]:
) -> AsyncIterator[VisualQARunner]:
runner = _new_runner(browser_config=browser_config, reporters=reporters)
try:
yield runner
Expand Down
2 changes: 1 addition & 1 deletion src/frontend_visualqa/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def _load_class(name: str) -> Any:
return loaded


def _resolve_default_visualize(claim_verifier: Any, fallback: bool) -> bool:
def _resolve_default_visualize(claim_verifier: ClaimVerifier, fallback: bool) -> bool:
"""Read *claim_verifier*'s ``_visualize`` flag, falling back to *fallback* if absent.

Shared by construction and ``_rebind_claim_verifier`` so the two places that
Expand Down