Skip to content

Commit 468fce7

Browse files
committed
test: drop imports orphaned by the MockClient consolidation
Removing the three local mock classes left their type-only imports (AsyncIterator, LLMResponse, ChunkType, StreamChunk, ToolSpec) unused; drop the ones no longer referenced.
1 parent 71dc68e commit 468fce7

2 files changed

Lines changed: 1 addition & 5 deletions

File tree

tests/unit/test_eval_budget.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,13 @@
22

33
from __future__ import annotations
44

5-
from collections.abc import AsyncIterator
65
from functools import partial
76
from typing import Any
87

98
import pytest
109

11-
from forge.clients.base import ChunkType, StreamChunk
1210
from forge.context.strategies import TieredCompact
13-
from forge.core.workflow import LLMResponse, ToolCall, ToolSpec, TextResponse
11+
from forge.core.workflow import ToolCall, TextResponse
1412

1513
from tests.conftest import MockClient
1614
from tests.eval.eval_runner import EvalConfig, RunResult, run_scenario

tests/unit/test_runner.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
from __future__ import annotations
44

55
import asyncio
6-
from collections.abc import AsyncIterator
76
from unittest.mock import MagicMock
87

98
import pytest
@@ -15,7 +14,6 @@
1514
from forge.core.runner import WorkflowRunner
1615
from pydantic import BaseModel
1716
from forge.core.workflow import (
18-
LLMResponse,
1917
TextResponse,
2018
ToolCall,
2119
ToolDef,

0 commit comments

Comments
 (0)