tomwer 1.1.0rc7__py3-none-any.whl → 1.2.0a0__py3-none-any.whl

tomwer/gui/reconstruction/axis/radioaxis.py

@@ -30,33 +30,35 @@ __authors__ = ["C. Nemoz", "H. Payno"]
 __license__ = "MIT"
 __date__ = "25/02/2019"
 
-from bisect import bisect_left
-from silx.gui import qt
-import logging
-from tomwer.core.utils import image
-import numpy
 import enum
+import logging
 import os
-from tomwer.core.scan.scanbase import TomwerScanBase
-from tomwer.core.scan.scanfactory import ScanFactory
-from tomwer.gui.utils.qt_utils import block_signals
-from tomwer.synctools.axis import QAxisRP
+from bisect import bisect_left
+from typing import Optional, Union
+
+import numpy
+import scipy.signal
+from silx.gui import qt
+from silx.io.url import DataUrl
+from silx.utils.deprecation import deprecated
+
 from tomwer.core.process.reconstruction.axis import mode as axis_mode
 from tomwer.core.process.reconstruction.axis.anglemode import CorAngleMode
 from tomwer.core.process.reconstruction.axis.params import (
-    AxisCalculationInput,
-    DEFAULT_CMP_THETA,
     DEFAULT_CMP_N_SUBSAMPLING_Y,
     DEFAULT_CMP_OVERSAMPLING,
     DEFAULT_CMP_TAKE_LOG,
+    DEFAULT_CMP_THETA,
+    AxisCalculationInput,
 )
-from silx.io.url import DataUrl
-import scipy.signal
-from .CompareImages import CompareImages
+from tomwer.core.scan.scanbase import TomwerScanBase
+from tomwer.core.scan.scanfactory import ScanFactory
+from tomwer.core.utils import image
 from tomwer.gui.utils.buttons import PadlockButton
-from silx.utils.deprecation import deprecated
-from typing import Union
-from typing import Optional
+from tomwer.gui.utils.qt_utils import block_signals
+from tomwer.synctools.axis import QAxisRP
+
+from .CompareImages import CompareImages
 
 _logger = logging.getLogger(__name__)
 
@@ -283,8 +285,7 @@ class RadioAxisWindow(qt.QMainWindow):
                 raise ValueError("Fail to discover a valid scan in %s" % scan)
         elif not isinstance(_scan, TomwerScanBase):
             raise ValueError(
-                "type of %s (%s) is invalid, scan should be a "
-                "file/dir path or an instance of ScanBase" % (scan, type(scan))
+                f"type of {scan} ({type(scan)}) is invalid, scan should be a file/dir path or an instance of ScanBase"
             )
         assert isinstance(_scan, TomwerScanBase)
 
@@ -376,7 +377,7 @@ class RadioAxisWindow(qt.QMainWindow):
         if _imgA.shape != _imgB.shape:
             _logger.error(
                 "The two provided images have incoherent shapes "
-                "(%s vs %s)" % (_imgA.shape, _imgB.shape)
+                f"({_imgA.shape} vs {_imgB.shape})"
             )
         elif _imgA.ndim != 2:
             _logger.error("Image shape are not 2 dimensional")
@@ -1038,7 +1039,7 @@ class _ROIDefinition(qt.QWidget):
                 self._xOriginSB.setValue(x_origin)
                 self._yOriginSB.setValue(y_origin)
             except Exception:
-                _logger.warning("unable to determine origin for {}".format(str(scan)))
+                _logger.warning(f"unable to determine origin for {scan}")
 
 
 @enum.unique
@@ -1547,8 +1548,7 @@ class AxisTabWidget(qt.QTabWidget):
         current_mode = self.getMode()
         if current_mode not in axis_mode._VALID_INPUTS:
             raise ValueError(
-                "valid input should have been defined for the "
-                "mode {}".format(current_mode.value)
+                f"valid input should have been defined for the mode {current_mode.value}"
             )
         else:
             valid_inputs = axis_mode._VALID_INPUTS[current_mode]
@@ -2017,16 +2017,16 @@ class _SliceSelector(qt.QWidget):
         else:
             return self._otherSB.value()
 
-    def setSlice(self, slice):
-        if slice is None:
+    def setSlice(self, slice_):
+        if slice_ is None:
             return
-        if slice == "middle":
+        if slice_ == "middle":
             idx = self._modeCB.findText("middle")
             self._modeCB.setCurrentIndex(idx)
         else:
             idx = self._modeCB.findText("other")
             self._modeCB.setCurrentIndex(idx)
-            self._otherSB.setValue(slice)
+            self._otherSB.setValue(slice_)
         self.sigChanged.emit()
 
     def getMode(self):
@@ -2236,15 +2236,13 @@ class _InputWidget(qt.QWidget):
                 axis_mode._InputType.from_value(mode)
             except Exception:
                 raise ValueError(
-                    "mode {} should be an instance of {}".format(
-                        mode, axis_mode._InputType
-                    )
+                    f"mode {mode} should be an instance of {axis_mode._InputType}"
                 )
         if len(modes) == 2:
             self._sinogramGB.setEnabled(True)
             self._radioGB.setEnabled(True)
         elif len(modes) > 2:
-            raise ValueError("invalid input {}".format(modes))
+            raise ValueError(f"invalid input {modes}")
         elif len(modes) < 0:
             raise ValueError("modes is empty")
         else:
@@ -2537,7 +2535,7 @@ class _ManualFramesSelection(qt.QWidget):
 
     @staticmethod
     def _angleToStr(angle: float) -> str:
-        return "{:0.3f}".format(float(angle))
+        return f"{float(angle):0.3f}"
 
     def setScan(self, scan: Union[TomwerScanBase, None]) -> None:
         self._anglesAvailable.clear()

tomwer/gui/reconstruction/darkref/darkrefcopywidget.py

@@ -31,14 +31,15 @@ __date__ = "13/08/2021"
 import logging
 import shutil
 from typing import Union
+
 from silx.gui import qt
+from silx.gui.dialog.DataFileDialog import DataFileDialog
+from silx.io.url import DataUrl
+
+from tomwer.core import settings, utils
 from tomwer.core.process.reconstruction.darkref.darkrefscopy import DarkRefsCopy
-from tomwer.io.utils import get_default_directory
-from tomwer.core import settings
-from tomwer.core import utils
 from tomwer.gui.reconstruction.darkref.darkrefwidget import DarkRefWidget
-from silx.io.url import DataUrl
-from silx.gui.dialog.DataFileDialog import DataFileDialog
+from tomwer.io.utils import get_default_directory
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/darkref/darkrefwidget.py

@@ -28,10 +28,12 @@ __license__ = "MIT"
 __date__ = "10/01/2018"
 
 
-from silx.gui import qt
 import logging
-from tomwer.core.process.reconstruction.darkref.darkrefs import DarkRefs
+
+from silx.gui import qt
+
 from tomwer.core.process.reconstruction.darkref import params as dkrf
+from tomwer.core.process.reconstruction.darkref.darkrefs import DarkRefs
 from tomwer.core.scan.scanbase import TomwerScanBase
 from tomwer.gui.utils.sandboxes import RegularExpressionSandBox
 from tomwer.synctools.darkref import QDKRFRP

tomwer/gui/reconstruction/lamino/tofu/misc.py

@@ -28,10 +28,12 @@ __license__ = "MIT"
 __date__ = "01/06/2018"
 
 
+import logging
+
 from silx.gui import qt
+
 from tomwer.gui import icons
 from tomwer.gui.utils.buttons import PadlockButton
-import logging
 
 _logger = logging.getLogger()
 

tomwer/gui/reconstruction/lamino/tofu/projections.py

@@ -28,19 +28,27 @@ __license__ = "MIT"
 __date__ = "01/06/2018"
 
 
+import logging
 import os
+
 from silx.gui import qt
-import logging
-from tomwer.core.process.reconstruction.lamino.tofu import getDark, getFlats, FFCWhen
-from tomwer.core.utils import getDARK_N, getDim1Dim2
-from tomwer.core.utils import getParametersFromParOrInfo
-from tomwer.core.utils import getTomo_N, getPixelSize, getClosestEnergy
+
+from tomwer.core.process.reconstruction.lamino.tofu import FFCWhen, getDark, getFlats
+from tomwer.core.utils import (
+    getClosestEnergy,
+    getDARK_N,
+    getDim1Dim2,
+    getParametersFromParOrInfo,
+    getPixelSize,
+    getTomo_N,
+)
 from tomwer.core.utils.char import PSI_CHAR, THETA_CHAR
 from tomwer.gui import icons
-from tomwer.gui.utils.illustrations import _IllustrationWidget, _IllustrationDialog
+from tomwer.gui.utils.illustrations import _IllustrationDialog, _IllustrationWidget
 from tomwer.gui.utils.unitsystem import MetricEntry
-from .TofuOptionLoader import _TofuOptionLoader, _getterSetter
-from .misc import _AngleWidget, PadlockButton
+
+from .misc import PadlockButton, _AngleWidget
+from .TofuOptionLoader import _getterSetter, _TofuOptionLoader
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/lamino/tofu/tofu.py

@@ -30,21 +30,24 @@ __license__ = "MIT"
 __date__ = "01/06/2018"
 
 
+import logging
+
 from silx.gui import qt
 from silx.io import dictdump
-import logging
+
 from tomwer.core.process.reconstruction.lamino.tofu import (
     SCAN_TYPES,
     SCAN_TYPES_I,
     FFCWhen,
+    _retrieve_opts_recons_cmd,
 )
-from tomwer.core.process.reconstruction.lamino.tofu import _retrieve_opts_recons_cmd
 from tomwer.core.utils.char import PSI_CHAR
 from tomwer.gui import icons
 from tomwer.gui.reconstruction.lamino.tofu import settings
-from .TofuOptionLoader import _TofuOptionLoader, _getterSetter
+
 from .projections import InputProjectionsScrollArea
 from .tofuexpert import ExpertTofuWidget
+from .TofuOptionLoader import _getterSetter, _TofuOptionLoader
 from .tofuoutput import OutputTofuWidget
 
 _logger = logging.getLogger(__name__)

tomwer/gui/reconstruction/lamino/tofu/tofuexpert.py

@@ -28,12 +28,15 @@ __license__ = "MIT"
 __date__ = "01/06/2018"
 
 
+import logging
 import subprocess
+
 from silx.gui import qt
+
 import tomwer.core.utils.gpu
 from tomwer.core.process.reconstruction.lamino.tofu import has_tofu
-import logging
-from .TofuOptionLoader import _TofuOptionLoader, _getterSetter
+
+from .TofuOptionLoader import _getterSetter, _TofuOptionLoader
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/lamino/tofu/tofuoutput.py

@@ -28,21 +28,24 @@ __license__ = "MIT"
 __date__ = "01/06/2018"
 
 
+import logging
 import os
+
 from silx.gui import qt
-import logging
+
 from tomwer.core.process.reconstruction.lamino.tofu import (
-    SLICE_STACK_TYPE,
-    ROTATION_CENTER_TYPE,
     LAMINO_ANGLE_TYPE,
     PSI_ANGLE_TYPE,
+    ROTATION_CENTER_TYPE,
+    SLICE_STACK_TYPE,
 )
 from tomwer.core.settings import get_lbsram_path
 from tomwer.core.utils.char import DELTA_CHAR
 from tomwer.gui.reconstruction.lamino.tofu import settings
 from tomwer.gui.utils.illustrations import _IllustrationWidget
-from .TofuOptionLoader import _TofuOptionLoader, _getterSetter
-from .misc import _AngleWidget, _RegionLE, PadlockButton
+
+from .misc import PadlockButton, _AngleWidget, _RegionLE
+from .TofuOptionLoader import _getterSetter, _TofuOptionLoader
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/nabu/action.py

@@ -29,6 +29,7 @@ __date__ = "04/08/2020"
 
 
 from silx.gui import qt
+
 from tomwer.gui import icons as tomwer_icons
 
 

tomwer/gui/reconstruction/nabu/castvolume.py

@@ -28,18 +28,20 @@ __license__ = "MIT"
 __date__ = "15/12/2021"
 
 
+import logging
+from typing import Optional
+
 from silx.gui import qt
+from silx.gui.utils import blockSignals
+
 from tomwer.core.process.reconstruction.nabu.castvolume import (
+    DEFAULT_OUTPUT_DIR,
     RESCALE_MAX_PERCENTILE,
     RESCALE_MIN_PERCENTILE,
-    DEFAULT_OUTPUT_DIR,
 )
 from tomwer.core.process.reconstruction.nabu.nabucommon import NabuOutputFileFormat
 from tomwer.gui.qlefilesystem import QLFileSystem
 from tomwer.gui.reconstruction.nabu.nabuconfig.output import QNabuFileFormatComboBox
-from silx.gui.utils import blockSignals
-import logging
-from typing import Optional
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/nabu/check.py

@@ -28,10 +28,11 @@ __license__ = "MIT"
 __date__ = "02/12/2021"
 
 
+from silx.gui import qt
+from tomoscan.esrf.scan.hdf5scan import ImageKey
+
 from tomwer.core.scan.hdf5scan import HDF5TomoScan
 from tomwer.core.utils.nxtomoutils import get_n_series
-from tomoscan.esrf.scan.hdf5scan import ImageKey
-from silx.gui import qt
 
 
 def check_dark_series(scan: HDF5TomoScan, logger=None, user_input: bool = True) -> bool:

tomwer/gui/reconstruction/nabu/nabuconfig/base.py

@@ -29,8 +29,11 @@ __date__ = "11/02/2020"
 
 
 from typing import Union
-from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
-from tomwer.core.process.reconstruction.nabu.utils import ConfigurationLevel
+
+from tomwer.core.process.reconstruction.nabu.utils import (
+    ConfigurationLevel,
+    _NabuStages,
+)
 
 
 class _FilteringObject:

tomwer/gui/reconstruction/nabu/nabuconfig/ctf.py

@@ -1,10 +1,13 @@
-from silx.gui import qt
 from typing import Optional, Union
+
+from silx.gui import qt
+from silx.gui.utils import blockSignals
+from silx.utils.enum import Enum as _Enum
+
 from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
 from tomwer.gui.reconstruction.nabu.nabuconfig import base
 from tomwer.gui.utils.illustrations import _IllustrationWidget
-from silx.gui.utils import blockSignals
-from silx.utils.enum import Enum as _Enum
+
 from .reconstruction import AnglesFileWidget as Filewidget
 
 

tomwer/gui/reconstruction/nabu/nabuconfig/nabuconfig.py

@@ -29,15 +29,16 @@ __date__ = "11/02/2020"
 
 
 from silx.gui import qt
+
+from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
+from tomwer.gui.reconstruction.nabu.nabuconfig.output import _NabuOutputConfig
+from tomwer.gui.reconstruction.nabu.nabuconfig.phase import _NabuPhaseConfig
 from tomwer.gui.reconstruction.nabu.nabuconfig.preprocessing import (
     _NabuPreProcessingConfig,
 )
-from tomwer.gui.reconstruction.nabu.nabuconfig.phase import _NabuPhaseConfig
 from tomwer.gui.reconstruction.nabu.nabuconfig.reconstruction import (
     _NabuReconstructionConfig,
 )
-from tomwer.gui.reconstruction.nabu.nabuconfig.output import _NabuOutputConfig
-from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
 
 
 class NabuConfiguration(qt.QWidget):

tomwer/gui/reconstruction/nabu/nabuconfig/output.py

@@ -28,14 +28,16 @@ __license__ = "MIT"
 __date__ = "11/02/2020"
 
 
+import logging
 import os
 import typing
+
 from silx.gui import qt
+
 from tomwer.core.process.reconstruction.nabu.nabucommon import NabuOutputFileFormat
+from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
 from tomwer.gui.qlefilesystem import QLFileSystem
 from tomwer.gui.reconstruction.nabu.nabuconfig.base import _NabuStageConfigBase
-from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
-import logging
 from tomwer.io.utils import get_default_directory
 
 try:
@@ -145,12 +147,12 @@ class NabuOutputLocationWidget(qt.QWidget):
         else:
             return self._outputDirQLE.text()
 
-    def setOutputDir(self, dir):
-        if dir in (None, ""):
+    def setOutputDir(self, output_dir):
+        if output_dir in (None, ""):
             self._defaultOutput.setChecked(True)
         else:
             self._defaultOutput.setChecked(False)
-            self._outputDirQLE.setText(dir)
+            self._outputDirQLE.setText(output_dir)
 
 
 class _NabuOutputConfig(_NabuStageConfigBase, qt.QWidget):

tomwer/gui/reconstruction/nabu/nabuconfig/phase.py

@@ -28,17 +28,19 @@ __license__ = "MIT"
 __date__ = "11/02/2020"
 
 
+import logging
 import typing
-from tomwer.utils import docstring
+
 from silx.gui import qt
-from tomwer.gui.utils.scrollarea import QComboBoxIgnoreWheel as QComboBox
+from silx.utils.enum import Enum as _Enum
+
+from tomwer.core.process.reconstruction.nabu.utils import _NabuPhaseMethod, _NabuStages
+from tomwer.core.utils.char import BETA_CHAR, DELTA_CHAR
 from tomwer.gui.reconstruction.nabu.nabuconfig import base
 from tomwer.gui.reconstruction.nabu.nabuconfig.ctf import CTFConfig
 from tomwer.gui.utils.inputwidget import SelectionLineEdit
-from tomwer.core.utils.char import BETA_CHAR, DELTA_CHAR
-from tomwer.core.process.reconstruction.nabu.utils import _NabuStages, _NabuPhaseMethod
-import logging
-from silx.utils.enum import Enum as _Enum
+from tomwer.gui.utils.scrollarea import QComboBoxIgnoreWheel as QComboBox
+from tomwer.utils import docstring
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/nabu/nabuconfig/preprocessing.py

@@ -28,19 +28,19 @@ __license__ = "MIT"
 __date__ = "06/12/2021"
 
 
+import logging
 from typing import Optional
+
 from silx.gui import qt
-from tomwer.gui.reconstruction.nabu.nabuconfig.base import _NabuStageConfigBase
-from tomwer.utils import docstring
+
 from tomwer.core.process.reconstruction.nabu.utils import (
     _NabuStages,
     _RingCorrectionMethod,
 )
-from tomwer.gui.utils.scrollarea import (
-    QComboBoxIgnoreWheel as QComboBox,
-    QDoubleSpinBoxIgnoreWheel as QDoubleSpinBox,
-)
-import logging
+from tomwer.gui.reconstruction.nabu.nabuconfig.base import _NabuStageConfigBase
+from tomwer.gui.utils.scrollarea import QComboBoxIgnoreWheel as QComboBox
+from tomwer.gui.utils.scrollarea import QDoubleSpinBoxIgnoreWheel as QDoubleSpinBox
+from tomwer.utils import docstring
 
 _logger = logging.getLogger(__name__)
 

tomwer/gui/reconstruction/nabu/nabuconfig/reconstruction.py

@@ -27,26 +27,29 @@ __authors__ = ["H. Payno"]
 __license__ = "MIT"
 __date__ = "12/02/2020"
 
-import typing
 import functools
-from tomwer.utils import docstring
+import logging
+import typing
+
 from silx.gui import qt
+from silx.gui.dialog.DataFileDialog import DataFileDialog
+from silx.utils.enum import Enum as _Enum
+
+from tomwer.core.process.reconstruction.nabu.nabuslices import NabuSliceMode
 from tomwer.core.process.reconstruction.nabu.utils import (
-    _NabuStages,
-    _NabuReconstructionMethods,
     _NabuFBPFilterType,
     _NabuPaddingType,
+    _NabuReconstructionMethods,
+    _NabuStages,
 )
-from tomwer.core.process.reconstruction.nabu.nabuslices import NabuSliceMode
 from tomwer.gui.reconstruction.nabu.nabuconfig.base import _NabuStageConfigBase
 from tomwer.gui.utils.inputwidget import SelectionLineEdit
-from tomwer.gui.utils.scrollarea import QComboBoxIgnoreWheel
-from tomwer.gui.utils.scrollarea import QDoubleSpinBoxIgnoreWheel
-from tomwer.gui.utils.scrollarea import QSpinBoxIgnoreWheel
-from silx.utils.enum import Enum as _Enum
-from silx.gui.dialog.DataFileDialog import DataFileDialog
-
-import logging
+from tomwer.gui.utils.scrollarea import (
+    QComboBoxIgnoreWheel,
+    QDoubleSpinBoxIgnoreWheel,
+    QSpinBoxIgnoreWheel,
+)
+from tomwer.utils import docstring
 
 _logger = logging.getLogger(__name__)
 
@@ -543,15 +546,15 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         else:
             return None
 
-    def setFBPFilterType(self, filter):
-        if type(filter) is str and filter.lower() == "none":
-            filter = None
-        if filter is None:
+    def setFBPFilterType(self, filter_type):
+        if type(filter_type) is str and filter_type.lower() == "none":
+            filter_type = None
+        if filter_type is None:
             self._fbpFilterCB.setChecked(False)
         else:
             self._fbpFilterCB.setChecked(True)
-            filter = _NabuFBPFilterType(filter)
-            filter_index = self._fbpFilterType.findText(filter.value)
+            filter_type = _NabuFBPFilterType(filter_type)
+            filter_index = self._fbpFilterType.findText(filter_type.value)
             self._fbpFilterType.setCurrentIndex(filter_index)
 
     def getFBPPaddingType(self) -> _NabuPaddingType:
@@ -778,33 +781,33 @@ class _SubRegionEditor(qt.QObject):
         else:
             return None
 
-    def setSubRegionMin(self, min: typing.Union[None, int]) -> None:
+    def setSubRegionMin(self, sub_region_min: typing.Union[None, int]) -> None:
         """
 
         :param min: if min is None or -1 wr will expand it to - infinity
         """
-        if min in (-1, 0):
-            min = None
-        if min is None:
+        if sub_region_min in (-1, 0):
+            sub_region_min = None
+        if sub_region_min is None:
             self._minCB.setChecked(False)
         else:
             self._minCB.setChecked(True)
-            self._minQLE.setText(str(min))
+            self._minQLE.setText(str(sub_region_min))
 
-    def setSubRegionMax(self, max) -> None:
+    def setSubRegionMax(self, sub_region_max) -> None:
         """
 
         :param max: if max is None or -1 wr will expand it to infinity
         """
-        if type(max) is str:
-            max = int(max)
-        if max == -1:
-            max = None
-        if max is None:
+        if type(sub_region_max) is str:
+            sub_region_max = int(sub_region_max)
+        if sub_region_max == -1:
+            sub_region_max = None
+        if sub_region_max is None:
             self._maxCB.setChecked(False)
         else:
             self._maxCB.setChecked(True)
-            self._maxQLE.setText(str(max))
+            self._maxQLE.setText(str(sub_region_max))
 
     def getSubRegion(self) -> tuple:
         """
@@ -814,14 +817,18 @@ class _SubRegionEditor(qt.QObject):
         """
         return self.getSubRegionMin(), self.getSubRegionMax()
 
-    def setSubRegion(self, min: typing.Union[None, int], max: typing.Union[None, int]):
+    def setSubRegion(
+        self,
+        sub_region_min: typing.Union[None, int],
+        sub_region_max: typing.Union[None, int],
+    ):
         """
 
         :param tuple min:
         :param tuple max:
         """
-        self.setSubRegionMin(min)
-        self.setSubRegionMax(max)
+        self.setSubRegionMin(sub_region_min)
+        self.setSubRegionMax(sub_region_max)
 
     def _signalConfChanged(self, param):
         self.sigConfChanged.emit(param)

tomwer/gui/reconstruction/nabu/nabuflow.py

@@ -27,14 +27,17 @@ __authors__ = ["H. Payno"]
 __license__ = "MIT"
 __date__ = "04/03/2019"
 
+import functools
+import typing
+
 from silx.gui import qt
+
+from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
 from tomwer.gui import icons
-from ...utils.flow import FlowDirection, FlowCanvas
 from tomwer.gui.utils.illustrations import _IllustrationWidget
 from tomwer.utils import docstring
-from tomwer.core.process.reconstruction.nabu.utils import _NabuStages
-import functools
-import typing
+
+from ...utils.flow import FlowCanvas, FlowDirection
 
 
 class NabuFlowControl(qt.QWidget):