[aDAG] Fix test_accelerated_dag regression

Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
ray-project · Sep 6, 2024 · 12e72f3 · 12e72f3
1 parent 5c70d96
commit 12e72f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/ray/dag/tests/experimental/test_accelerated_dag.py b/python/ray/dag/tests/experimental/test_accelerated_dag.py
@@ -1849,6 +1849,8 @@ def test_event_profiling(ray_start_regular, monkeypatch):
         assert event.method_name == "inc"
         assert event.operation in ["READ", "COMPUTE", "WRITE"]
 
+    adag.teardown()
+
 
 @ray.remote
 class TestWorker: