Checking whether Unittest issues are related to catch_interrupts import.

qutech · Feb 7, 2024 · 914372c · 914372c
1 parent 46d45ff
commit 914372c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/qumada/measurement/doNd_enhanced/doNd_enhanced.py b/src/qumada/measurement/doNd_enhanced/doNd_enhanced.py
@@ -41,12 +41,16 @@
 from qcodes.dataset.descriptions.detect_shapes import detect_shape_of_measurement
 from qcodes.dataset.descriptions.versioning.rundescribertypes import Shapes
 from qcodes.dataset.dond.do_nd_utils import (
-    _catch_interrupts,
+    # _catch_interrupts,
     _handle_plotting,
     _register_actions,
     _register_parameters,
     _set_write_period,
 )
+try:
+    from qcodes.dataset.dond.do_nd_utils import _catch_interrupts
+except ImportError:
+    from qcodes.dataset.dond.do_nd_utils import catch_interrupts as _catch_interrupts
 from qcodes.dataset.experiment_container import Experiment
 from qcodes.dataset.measurements import Measurement
 from qcodes.dataset.threading import (  # SequentialParamsCaller,; ThreadPoolParamsCaller,