Skip to content

Commit b850762

Browse files
committed
undo multiprocessing test option bc it may be too slow for the ci's. (at least they pass)
1 parent ac09868 commit b850762

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tests/utils/test_parallelize.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,8 @@ def func(request) -> Callable:
6666
# in case of failure.
6767

6868

69-
@pytest.mark.parametrize("backend", ["loky", "multiprocessing"])
7069
@pytest.mark.timeout(30)
71-
def test_parallelize(func, backend):
70+
def test_parallelize_loky(func):
7271
seed = 42
7372
n = 8
7473
n_jobs = 2
@@ -80,7 +79,7 @@ def test_parallelize(func, backend):
8079
init_threads = numba.get_num_threads()
8180
expected = np.vstack([func(a1, arr2, check_threads=False) for a1 in arr1])
8281

83-
p_func = parallelize(runner, arr1, n_jobs=n_jobs, backend=backend, use_ixs=False, extractor=np.vstack)
82+
p_func = parallelize(runner, arr1, n_jobs=n_jobs, backend="loky", use_ixs=False, extractor=np.vstack)
8483
result = p_func(arr2)
8584

8685
assert numba.get_num_threads() == init_threads, "Number of threads should stay the same after parallelization"

0 commit comments

Comments
 (0)