Skip to content
Closed
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
10 changes: 9 additions & 1 deletion tests/unit/agents/flash/test_turn_index_snapshot.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,15 @@ def mock_context():


def _runner(mock_context, observations):
with patch("artemis.controllers.unified_controller.get_driver"):
# FlashRunner.__init__ builds a VisualStepSummarizer, whose constructor
# resolves a real model client and raises without GOOGLE_API_KEY. Setting
# `runner.summarizer = None` below is too late -- the object is already
# built by then -- so the class is patched for the construction itself.
# These tests never summarize anything; they assert index resolution.
with (
patch("artemis.controllers.unified_controller.get_driver"),
patch("artemis.agents.flash.runner.VisualStepSummarizer"),
):
runner = FlashRunner(mock_context, goal="Open Wi-Fi and Display")
runner.summarizer = None
runner.executor._session = _FakeSession(observations)
Expand Down