hiro/two_qubit_model
1
0
Fork 0
mirror of https://github.com/vale981/two_qubit_model synced 2025-03-04 17:21:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix killing the integration

Browse source
This commit is contained in:
Valentin Boettcher 2022-12-01 12:27:32 -05:00
parent cda74a1e70
commit 5bd24d10df
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 17 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
hiro_models/model_auxiliary.py
View file

@ -21,6 +21,8 @@ import copy
import os
import numpy as np
from multiprocessing import Process
import hops.core.signal_delay as signal_delay
import signal
@contextmanager
@ -155,10 +157,21 @@ def integrate(
analysis_process.start()
logging.info(f"Started analysis process with pid {analysis_process.pid}.")
if single_process:
supervisor.integrate_single_process(clear_pd)
else:
supervisor.integrate(clear_pd)
def signal_handler(_):
del _
if analysis_process is not None:
analysis_process.kill()
analysis_process.join()
with signal_delay.sig_delay(
[signal.SIGINT, signal.SIGTERM, signal.SIGHUP, signal.SIGUSR1],
signal_handler,
):
if single_process:
supervisor.integrate_single_process(clear_pd)
else:
supervisor.integrate(clear_pd)
if analysis_process:
analysis_process.join()