perf(Ensemble): Replaced python built-in multiprocessing to pathos mu…

…ltiprocessing
frankkramer-lab · Jun 13, 2022 · c79110f · c79110f
1 parent 018aed8
commit c79110f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/aucmedi/ensemble/bagging.py b/aucmedi/ensemble/bagging.py
@@ -23,7 +23,7 @@
 import os
 import tempfile
 from tensorflow.keras.callbacks import ModelCheckpoint, CSVLogger
-import multiprocessing as mp
+from pathos.helpers import mp   # instead of 'import multiprocessing as mp'
 import numpy as np
 import shutil
 # Internal libraries

diff --git a/aucmedi/ensemble/composite.py b/aucmedi/ensemble/composite.py
@@ -23,7 +23,7 @@
 import os
 import tempfile
 from tensorflow.keras.callbacks import ModelCheckpoint, CSVLogger
-import multiprocessing as mp
+from pathos.helpers import mp   # instead of 'import multiprocessing as mp'
 import numpy as np
 import shutil
 # Internal libraries

diff --git a/aucmedi/ensemble/stacking.py b/aucmedi/ensemble/stacking.py
@@ -23,7 +23,7 @@
 import os
 import tempfile
 from tensorflow.keras.callbacks import ModelCheckpoint, CSVLogger
-import multiprocessing as mp
+from pathos.helpers import mp   # instead of 'import multiprocessing as mp'
 import numpy as np
 import shutil
 # Internal libraries