IPKM/mdevaluate
6
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Readded the try except block to shifted_correlation.

Browse Source
This commit is contained in:
sebastiankloth 2023-06-27 17:29:36 +02:00
parent a1d0cee510
commit 8146a9b270
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mdevaluate/correlation.py
View File

@ -192,13 +192,22 @@ def shifted_correlation(
) )
else: else:
pool = ProcessPool(nodes=nodes) pool = ProcessPool(nodes=nodes)
result = np.array( try:
pool.map(partial(get_correlation, idx=idx, selector=selector), start_frames) result = np.array(
) pool.map(
pool.terminate() partial(get_correlation, idx=idx, selector=selector), start_frames
pool.restart() )
)
except Exception:
logger.warning(
"WARNING: Something went wrong during multiprocessing "
"in shifted_correlation."
)
finally:
pool.terminate()
pool.restart()
if average == True: if average:
clean_result = [] clean_result = []
for entry in result: for entry in result:
if np.all(entry == 0): if np.all(entry == 0):