test: Fix teardown

Signed-off-by: Weixin Deng <weixin@cs.washington.edu>

python/ray/dag/tests/experimental/test_torch_tensor_dag.py

@@ -1807,6 +1807,8 @@ def test_torch_tensor_nccl_all_reduce_scheduling(ray_start_regular):
     assert torch.equal(result[1], expected_tensor_value)
     assert result[2] == (value, shape, dtype)
 
+    compiled_dag.teardown()
+
 
 @pytest.mark.parametrize("ray_start_regular", [{"num_cpus": 4}], indirect=True)
 def test_torch_tensor_nccl_all_reduce_scheduling_one_ready_group(ray_start_regular):
@@ -1848,6 +1850,8 @@ def test_torch_tensor_nccl_all_reduce_scheduling_one_ready_group(ray_start_regul
     result = ray.get(ref)
     assert result == [(value * 2, shape, dtype) for _ in workers]
 
+    compiled_dag.teardown()
+
 
 if __name__ == "__main__":
     if os.environ.get("PARALLEL_CI"):