From 51f785e25b8dc2aaf05a8614f05a2faeffbdd169 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Fri, 13 Sep 2024 23:22:12 -0400 Subject: [PATCH] qxqxqx --- tests/unit_tests/test_server_client.py | 3 +++ tests/unit_tests/utils/tracer.py | 3 ++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/unit_tests/test_server_client.py b/tests/unit_tests/test_server_client.py index 32817d01..6a7ffed0 100644 --- a/tests/unit_tests/test_server_client.py +++ b/tests/unit_tests/test_server_client.py @@ -540,6 +540,9 @@ def test_foo_foo_bar_bar(sync_remote_runnable: RemoteRunnable) -> None: tracer = FakeTracer() assert sync_remote_runnable.batch([1], config={"callbacks": [tracer]}) == [2] assert len(tracer.runs) == 1 + # Child run exists from server side (this fails) + assert len(tracer.runs[0].child_runs[0]) == 1 # + assert tracer.runs[0].child_runs[0].name == "RunnableLambda" def test_batch(sync_remote_runnable: RemoteRunnable) -> None: diff --git a/tests/unit_tests/utils/tracer.py b/tests/unit_tests/utils/tracer.py index c1a41f78..4bbf1103 100644 --- a/tests/unit_tests/utils/tracer.py +++ b/tests/unit_tests/utils/tracer.py @@ -53,7 +53,8 @@ def _create_chain_run( ) -> Run: if name is None: # can't raise an exception from here, but can get a breakpoint - import pdb; pdb.set_trace() + # import pdb; pdb.set_trace() + pass return super()._create_chain_run( serialized, inputs,