From 5f9bd01fbeea877a3bee36d55ee116df959a6740 Mon Sep 17 00:00:00 2001 From: Saiyam Bharara Date: Tue, 20 Oct 2026 21:57:31 +0000 Subject: [PATCH] test(cuda.core): wait for the event-record target to turn pending Signed-off-by: Saiyam Bharara --- cuda_core/tests/graph/test_graph_node_update.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cuda_core/tests/graph/test_graph_node_update.py b/cuda_core/tests/graph/test_graph_node_update.py index 9f31df6a118..c846f230c26 100644 --- a/cuda_core/tests/graph/test_graph_node_update.py +++ b/cuda_core/tests/graph/test_graph_node_update.py @@ -120,7 +120,7 @@ def assert_exec_uses(graph, expected): graph.launch(stream) try: assert callback_started.wait(timeout=5) - assert expected.is_done is False + _wait_until(lambda: expected.is_done is False) unexpected = replacement if expected is original else original assert unexpected.is_done is True finally: