From f6d537b35f01568bbac8944e5258feafd89efb80 Mon Sep 17 00:00:00 2001 From: gtarpenning Date: Mon, 3 Mar 2025 19:28:14 -0800 Subject: [PATCH] fixtest --- tests/trace/test_weave_client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/trace/test_weave_client.py b/tests/trace/test_weave_client.py index 291f661423a1..d3a5970f37ea 100644 --- a/tests/trace/test_weave_client.py +++ b/tests/trace/test_weave_client.py @@ -1900,7 +1900,7 @@ def op_1(): client.flush(use_progress_bar=True) # make sure there are no pending jobs - assert client._get_pending_jobs() == (0, 0, 0, 0) + assert client._get_pending_jobs()["total_jobs"] == 0 assert client._has_pending_jobs() == False @@ -1912,13 +1912,13 @@ def op_1(): op_1() def fake_logger(status): - assert "main_jobs" in status + assert "job_counts" in status # flush with callback client.flush(callback=fake_logger) # make sure there are no pending jobs - assert client._get_pending_jobs() == (0, 0, 0, 0) + assert client._get_pending_jobs()["total_jobs"] == 0 assert client._has_pending_jobs() == False op_1() @@ -1927,5 +1927,5 @@ def fake_logger(status): client.flush(callback=fake_logger, use_progress_bar=True) # make sure there are no pending jobs - assert client._get_pending_jobs() == (0, 0, 0, 0) + assert client._get_pending_jobs()["total_jobs"] == 0 assert client._has_pending_jobs() == False