fix: minor test fix

qiboteam · Jan 24, 2025 · d905a65 · d905a65
1 parent 36844fa
commit d905a65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/qibojit/tests/test_backends.py b/src/qibojit/tests/test_backends.py
@@ -18,7 +18,9 @@ def test_thread_setter(backend):
     import numba
 
     original_threads = numba.get_num_threads()
+    backend.set_threads(1)
     assert numba.get_num_threads() == 1
+    backend.set_threads(original_threads)
 
 
 @pytest.mark.parametrize("array_type", [None, "float32", "float64"])