From b6412401c37e1f5e4f7dc190b7b34733d2f68932 Mon Sep 17 00:00:00 2001
From: Tom Augspurger <toaugspurger@nvidia.com>
Date: Tue, 4 Feb 2025 17:16:53 +0000
Subject: [PATCH] Use threads

---
 python/kvikio/tests/test_benchmarks.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/kvikio/tests/test_benchmarks.py b/python/kvikio/tests/test_benchmarks.py
index f69e3ee4f4..6707a86efc 100644
--- a/python/kvikio/tests/test_benchmarks.py
+++ b/python/kvikio/tests/test_benchmarks.py
@@ -28,7 +28,7 @@
         "zarr",
     ],
 )
-@pytest.mark.timeout(30)
+@pytest.mark.timeout(30, method="thread")
 def test_single_node_io(run_cmd, tmp_path, api):
     """Test benchmarks/single_node_io.py"""
 
@@ -60,7 +60,7 @@ def test_single_node_io(run_cmd, tmp_path, api):
         "posix",
     ],
 )
-@pytest.mark.timeout(30)
+@pytest.mark.timeout(30, method="thread")
 def test_zarr_io(run_cmd, tmp_path, api):
     """Test benchmarks/zarr_io.py"""
 
@@ -91,7 +91,7 @@ def test_zarr_io(run_cmd, tmp_path, api):
         "numpy",
     ],
 )
-@pytest.mark.timeout(30)
+@pytest.mark.timeout(30, method="thread")
 def test_http_io(run_cmd, api):
     """Test benchmarks/http_io.py"""
 
@@ -121,7 +121,7 @@ def test_http_io(run_cmd, api):
         "numpy",
     ],
 )
-@pytest.mark.timeout(30)
+@pytest.mark.timeout(30, method="thread")
 def test_s3_io(run_cmd, api):
     """Test benchmarks/s3_io.py"""