mypy

11 months ago · 4ba9c16f74
parent d1d7d0cb27
commit 4ba9c16f74
1 changed files with 7 additions and 6 deletions
--- a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py
+++ b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py
@ -2758,12 +2758,13 @@ def test_reprsentation_of_runnables() -> None:

    assert repr(RunnableLambda(func=f, afunc=af)) == "RunnableLambda(...)"

-    sequence = RunnableLambda(lambda x: x + 2) | {
-        "a": RunnableLambda(lambda x: x * 2),
-        "b": RunnableLambda(lambda x: x * 3),
-    }
-
-    assert repr(sequence) == (
+    assert repr(
+        RunnableLambda(lambda x: x + 2)
+        | {
+            "a": RunnableLambda(lambda x: x * 2),
+            "b": RunnableLambda(lambda x: x * 3),
+        }
+    ) == (
        "RunnableLambda(lambda x: x * 3)\n"
        "| {\n"
        "    a: RunnableLambda(...),\n"