diff --git a/src/scportrait/pipeline/segmentation/workflows.py b/src/scportrait/pipeline/segmentation/workflows.py index 2ad5d0cb..b677597f 100644 --- a/src/scportrait/pipeline/segmentation/workflows.py +++ b/src/scportrait/pipeline/segmentation/workflows.py @@ -15,7 +15,6 @@ from skimage.filters import median from skimage.morphology import binary_erosion, dilation, disk, erosion from skimage.segmentation import watershed -import _multiprocessing from scportrait.pipeline._utils.segmentation import ( contact_filter, @@ -1355,7 +1354,7 @@ def _check_gpu_status(self): cpu_id = int(cpu_name[cpu_name.find("-") + 1 :]) - 1 if cpu_id >= len(gpu_id_list): - cpu_id = cpu_id%current.n_processes + cpu_id = cpu_id % current.n_processes # track gpu_id and update GPU status self.gpu_id = gpu_id_list[cpu_id]