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

Removed except clause in shifted_correlation to always raise an exception.

Browse Source
This commit is contained in:
sebastiankloth 2023-06-30 08:58:36 +02:00
parent 2125c4de47
commit f453aa6fa0
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mdevaluate/correlation.py
View File

@ -192,17 +192,15 @@ def shifted_correlation(
) )
else: else:
pool = ProcessPool(nodes=nodes) pool = ProcessPool(nodes=nodes)
# Use try finally instead of a context manager to ensure the pool is
# restarted in case of working in a jupyter-notebook,
# otherwise the kernel has to be restarted.
try: try:
result = np.array( result = np.array(
pool.map( pool.map(
partial(get_correlation, idx=idx, selector=selector), start_frames partial(get_correlation, idx=idx, selector=selector), start_frames
) )
) )
except Exception:
logger.warning(
"WARNING: Something went wrong during multiprocessing "
"in shifted_correlation."
)
finally: finally:
pool.terminate() pool.terminate()
pool.restart() pool.restart()