Skip to content

Commit ea0b91c

Browse files
committed
add ruff lint on test
1 parent 86862b4 commit ea0b91c

1 file changed

Lines changed: 5 additions & 6 deletions

File tree

‎tests/test_multi_turn_integration.py‎

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,13 @@
88
5. Properly inform the model about available contexts/histories
99
"""
1010

11+
from unittest.mock import Mock, patch
12+
1113
import pytest
12-
from unittest.mock import patch, Mock
1314

14-
from rlm import RLM
15-
from rlm.core.types import UsageSummary, ModelUsageSummary
1615
import rlm.core.rlm as rlm_module
16+
from rlm import RLM
17+
from rlm.core.types import ModelUsageSummary, UsageSummary
1718

1819

1920
def create_mock_lm(responses: list[str]) -> Mock:
@@ -22,9 +23,7 @@ def create_mock_lm(responses: list[str]) -> Mock:
2223
mock.completion.side_effect = list(responses)
2324
mock.get_usage_summary.return_value = UsageSummary(
2425
model_usage_summaries={
25-
"mock": ModelUsageSummary(
26-
total_calls=1, total_input_tokens=100, total_output_tokens=50
27-
)
26+
"mock": ModelUsageSummary(total_calls=1, total_input_tokens=100, total_output_tokens=50)
2827
}
2928
)
3029
mock.get_last_usage.return_value = mock.get_usage_summary.return_value

0 commit comments

Comments
 (0)