diff --git a/libs/core/tests/unit_tests/runnables/test_config.py b/libs/core/tests/unit_tests/runnables/test_config.py index 397adf7b06c..7ffb407a1b5 100644 --- a/libs/core/tests/unit_tests/runnables/test_config.py +++ b/libs/core/tests/unit_tests/runnables/test_config.py @@ -228,13 +228,11 @@ def test_inherit_run_name_enabled() -> None: def test_inherit_run_name_with_chain() -> None: """Test inherit_run_name behavior in a chain of runnables.""" - from typing import List - from langchain_core.callbacks.base import BaseCallbackHandler from langchain_core.runnables import RunnableLambda # Track run names through callbacks - captured_names: List[str] = [] + captured_names: list[str] = [] class TestCallbackHandler(BaseCallbackHandler): def on_chain_start(self, serialized: dict, inputs: Any, **kwargs: Any) -> None: @@ -370,3 +368,4 @@ def test_inherit_run_name_merge_configs() -> None: "inherit_run_name should pass through ensure_config" ) assert ensured.get("run_name") == "test_run", "run_name should be preserved" +