more linting

Author: alexzhang13

tests/mock_lm.py

@@ -1,6 +1,7 @@
 """Mock LM for testing without a real model."""
 
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 from rlm.clients.base_lm import BaseLM
 from rlm.core.types import ModelUsageSummary, UsageSummary

tests/test_lm_handler.py

@@ -1,8 +1,8 @@
 """Tests for LMHandler using MockLM (no real LM required)."""
 
-from tests.mock_lm import MockLM
 from rlm.core.comms_utils import LMRequest, send_lm_request, send_lm_request_batched
 from rlm.core.lm_handler import LMHandler
+from tests.mock_lm import MockLM
 
 
 def test_lm_handler_single_request():