Skip to content

Commit eee35dd

Browse files
committed
Move multiprocessing config to image.py where it is actually used.
1 parent 154bfe7 commit eee35dd

2 files changed

Lines changed: 4 additions & 5 deletions

File tree

sourcefinder/image.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,16 @@
1616
from sourcefinder.utility import containers
1717
from sourcefinder.utility.uncertain import Uncertain
1818
import psutil
19-
from multiprocessing import Pool
19+
import multiprocessing
2020
from functools import cached_property
2121
from functools import partial
2222

2323
from scipy import ndimage
2424
from numba import guvectorize, float32, int32
2525
import os
2626

27+
multiprocessing.set_start_method("forkserver", force=True)
28+
2729
logger = logging.getLogger(__name__)
2830

2931

@@ -1205,7 +1207,7 @@ def _pyse(
12051207
)
12061208

12071209
if self.conf.image.allow_multiprocessing:
1208-
with Pool(psutil.cpu_count()) as p:
1210+
with multiprocessing.Pool(psutil.cpu_count()) as p:
12091211
fit_results = p.map(fit_islands_partial, island_list)
12101212
else:
12111213
fit_results = [fit_islands_partial(island) for island in

test/conftest.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,9 @@
11
import os
22
from pathlib import Path
33
import warnings
4-
import multiprocessing
54

65
DEFAULT = str((Path(__file__).parent / "data").absolute())
76
DATAPATH = os.environ.get("TKP_TESTPATH", DEFAULT)
87

9-
multiprocessing.set_start_method("forkserver", force=True)
10-
118
if not os.access(DATAPATH, os.X_OK):
129
warnings.warn(f"can't access {DATAPATH}")

0 commit comments

Comments
 (0)