ZH: Enable multiprocessing in reconstruction

This commit is contained in:
Eric Teunis de Boone 2023-01-12 13:39:06 +01:00
parent 3c99ac7118
commit 0ac1deff34

View file

@ -11,6 +11,7 @@ from mpl_toolkits.mplot3d import Axes3D # required for projection='3d' on old ma
import numpy as np
from os import path
import pickle
import joblib
from earsim import REvent
from atmocal import AtmoCal
@ -69,6 +70,7 @@ if __name__ == "__main__":
ev.antennas[i].t += total_clock_time
N_X, Xlow, Xhigh = 23, 100, 1200
with joblib.parallel_backend("loky"):
res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh)
## Save a pickle