Fix nb of threads

Author: youennmereljourdan

postprocess.py

@@ -40,7 +40,9 @@ def postproc():
     print(f"Mean result image written to [{mean_path}]")
 
     print(f"Computing all correlations between [{size}] results...")
-    correlations = postproc_srv.get_all_correlations(basedir, ids)
+    nb_cores = len(os.sched_getaffinity(0))
+    print(f"[{nb_cores}] cores available")
+    correlations = postproc_srv.get_all_correlations(basedir, ids, nb_cores)
     correlations.to_csv(corr_path, index=False, sep=';')
     print(f"Correlations written to [{corr_path}]")
 
@@ -53,7 +55,6 @@ def postproc():
 if __name__ == '__main__':
     now = datetime.now().strftime("%d-%m-%Y %H:%M:%S.%f")[:-3]
     print(f"Start [{now}]")
-    print(f"[{len(os.sched_getaffinity(0))}] cores available")
     postproc()
     now = datetime.now().strftime("%d-%m-%Y %H:%M:%S.%f")[:-3]
     print(f"End [{now}]")

postprocess/postprocess_service.py

@@ -44,16 +44,15 @@ def get_pairwise_correlation(self, src, tgt, path):
             jacc = self.corr_srv.get_correlation_coefficient(src_nii, tgt_nii, 'jaccard')
             return src, tgt, spear, pear, dice, jacc
 
-    def get_all_correlations(self, path, ids: List[str]) -> pd.DataFrame:
+    def get_all_correlations(self, path, ids: List[str], nb_cores: int) -> pd.DataFrame:
         ids.append('mean')
-        n = len(ids)
 
         args = []
         for i in range(n):
             for j in range(i, n):
                 args.append((ids[i], ids[j], path))
 
-        with Pool(processes=n) as pool:
+        with Pool(processes=nb_cores) as pool:
             results = pool.starmap(self.get_pairwise_correlation, args)
 
         data = []