diff --git a/sen2chain/multi_processing.py b/sen2chain/multi_processing.py index 71816b3f4383af6aba50a25b557e36d2a814d176..926c9075eba84ca404714a9dfb0d4c7b1421106f 100644 --- a/sen2chain/multi_processing.py +++ b/sen2chain/multi_processing.py @@ -4,6 +4,7 @@ import multiprocessing, subprocess import os from time import sleep import logging +from functools import partial #~ import psutil from .products import L1cProduct, L2aProduct @@ -34,14 +35,15 @@ def l2a_multiprocessing(process_list, nb_proc=4): pool.close() pool.join() -def multi_cldidx(l2a_identifier): +def multi_cldidx(indice_list, l2a_identifier): l2a = L2aProduct(l2a_identifier) l2a.process_cloud_mask_v2() - l2a.process_indices(["NDVI", "NDWIGAO", "NDWIMCF"], True, True) + l2a.process_indices(indice_list, True, True) -def cldidx_multiprocessing(process_list, nb_proc=4): +def cldidx_multiprocessing(process_list, indice_list=["NDVI", "NDWIGAO", "NDWIMCF"], nb_proc=4): """ """ pool = multiprocessing.Pool(nb_proc) - results = [pool.map(multi_cldidx, process_list)] + #~ function = partial(multi_cldidx, indice_list) + results = [pool.map(partial(multi_cldidx, indice_list), process_list)] pool.close() pool.join()