refactor(plots): plot_next_gen module renamed to plots

bec_widgets/examples/jupyter_console/jupyter_console_window.py

@@ -19,12 +19,12 @@ from bec_widgets.utils.widget_io import WidgetHierarchy as wh
 from bec_widgets.widgets.containers.dock import BECDockArea
 from bec_widgets.widgets.containers.layout_manager.layout_manager import LayoutManagerWidget
 from bec_widgets.widgets.editors.jupyter_console.jupyter_console import BECJupyterConsole
-from bec_widgets.widgets.plots_next_gen.image.image import Image
+from bec_widgets.widgets.plots.image.image import Image
-from bec_widgets.widgets.plots_next_gen.motor_map.motor_map import MotorMap
+from bec_widgets.widgets.plots.motor_map.motor_map import MotorMap
-from bec_widgets.widgets.plots_next_gen.multi_waveform.multi_waveform import MultiWaveform
+from bec_widgets.widgets.plots.multi_waveform.multi_waveform import MultiWaveform
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform import ScatterWaveform
+from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform import ScatterWaveform
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 
 class JupyterConsoleWindow(QWidget):  # pragma: no cover:

bec_widgets/qt_utils/side_panel.py

@@ -325,7 +325,7 @@ class ExampleApp(QMainWindow):  # pragma: no cover
         self.side_panel = SidePanel(self, orientation="left", panel_max_width=250)
         self.layout.addWidget(self.side_panel)
 
-        from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+        from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
         self.plot = Waveform()
         self.layout.addWidget(self.plot)

bec_widgets/widgets/containers/dock/dock_area.py

@@ -25,11 +25,11 @@ from bec_widgets.widgets.containers.dock.dock import BECDock, DockConfig
 from bec_widgets.widgets.control.device_control.positioner_box import PositionerBox
 from bec_widgets.widgets.control.scan_control.scan_control import ScanControl
 from bec_widgets.widgets.editors.vscode.vscode import VSCodeEditor
-from bec_widgets.widgets.plots_next_gen.image.image import Image
+from bec_widgets.widgets.plots.image.image import Image
-from bec_widgets.widgets.plots_next_gen.motor_map.motor_map import MotorMap
+from bec_widgets.widgets.plots.motor_map.motor_map import MotorMap
-from bec_widgets.widgets.plots_next_gen.multi_waveform.multi_waveform import MultiWaveform
+from bec_widgets.widgets.plots.multi_waveform.multi_waveform import MultiWaveform
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform import ScatterWaveform
+from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform import ScatterWaveform
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
 from bec_widgets.widgets.progress.ring_progress_bar.ring_progress_bar import RingProgressBar
 from bec_widgets.widgets.services.bec_queue.bec_queue import BECQueue
 from bec_widgets.widgets.services.bec_status_box.bec_status_box import BECStatusBox

bec_widgets/widgets/plots/image/image.py

@@ -14,14 +14,12 @@ from bec_widgets.qt_utils.error_popups import SafeProperty, SafeSlot
 from bec_widgets.qt_utils.toolbar import MaterialIconAction, SwitchableToolBarAction
 from bec_widgets.utils import ConnectionConfig
 from bec_widgets.utils.colors import Colors
-from bec_widgets.widgets.plots_next_gen.image.image_item import ImageItem
+from bec_widgets.widgets.plots.image.image_item import ImageItem
-from bec_widgets.widgets.plots_next_gen.image.toolbar_bundles.image_selection import (
+from bec_widgets.widgets.plots.image.toolbar_bundles.image_selection import (
     MonitorSelectionToolbarBundle,
 )
-from bec_widgets.widgets.plots_next_gen.image.toolbar_bundles.processing import (
+from bec_widgets.widgets.plots.image.toolbar_bundles.processing import ImageProcessingToolbarBundle
-    ImageProcessingToolbarBundle,
+from bec_widgets.widgets.plots.plot_base import PlotBase
-)
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/image/image_item.py

@@ -9,7 +9,7 @@ from pydantic import Field, ValidationError, field_validator
 from qtpy.QtCore import Signal
 
 from bec_widgets.utils import BECConnector, Colors, ConnectionConfig
-from bec_widgets.widgets.plots_next_gen.image.image_processor import (
+from bec_widgets.widgets.plots.image.image_processor import (
     ImageProcessor,
     ImageStats,
     ProcessingConfig,

bec_widgets/widgets/plots/image/image_plugin.py

@@ -4,7 +4,7 @@
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 
 from bec_widgets.utils.bec_designer import designer_material_icon
-from bec_widgets.widgets.plots_next_gen.image.image import Image
+from bec_widgets.widgets.plots.image.image import Image
 
 DOM_XML = """
 <ui language='c++'>
@@ -27,7 +27,7 @@ class ImagePlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
         return DOM_XML
 
     def group(self):
-        return "Plot Widgets Next Gen"
+        return "Plot Widgets"
 
     def icon(self):
         return designer_material_icon(Image.ICON_NAME)

bec_widgets/widgets/plots/image/register_image.py

@@ -6,7 +6,7 @@ def main():  # pragma: no cover
         return
     from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
 
-    from bec_widgets.widgets.plots_next_gen.image.image_plugin import ImagePlugin
+    from bec_widgets.widgets.plots.image.image_plugin import ImagePlugin
 
     QPyDesignerCustomWidgetCollection.addCustomWidget(ImagePlugin())
 

bec_widgets/widgets/plots/motor_map/motor_map.py

@@ -16,13 +16,11 @@ from bec_widgets.qt_utils.settings_dialog import SettingsDialog
 from bec_widgets.qt_utils.toolbar import MaterialIconAction
 from bec_widgets.utils import Colors, ConnectionConfig
 from bec_widgets.utils.colors import set_theme
-from bec_widgets.widgets.plots_next_gen.motor_map.settings.motor_map_settings import (
+from bec_widgets.widgets.plots.motor_map.settings.motor_map_settings import MotorMapSettings
-    MotorMapSettings,
+from bec_widgets.widgets.plots.motor_map.toolbar_bundles.motor_selection import (
-)
-from bec_widgets.widgets.plots_next_gen.motor_map.toolbar_bundles.motor_selection import (
     MotorSelectionToolbarBundle,
 )
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/motor_map/motor_map_plugin.py

@@ -4,7 +4,7 @@
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 
 from bec_widgets.utils.bec_designer import designer_material_icon
-from bec_widgets.widgets.plots_next_gen.motor_map.motor_map import MotorMap
+from bec_widgets.widgets.plots.motor_map.motor_map import MotorMap
 
 DOM_XML = """
 <ui language='c++'>
@@ -27,7 +27,7 @@ class MotorMapPlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
         return DOM_XML
 
     def group(self):
-        return "Plot Widgets Next Gen"
+        return "Plot Widgets"
 
     def icon(self):
         return designer_material_icon(MotorMap.ICON_NAME)

bec_widgets/widgets/plots/motor_map/register_motor_map.py

@@ -6,7 +6,7 @@ def main():  # pragma: no cover
         return
     from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
 
-    from bec_widgets.widgets.plots_next_gen.motor_map.motor_map_plugin import MotorMapPlugin
+    from bec_widgets.widgets.plots.motor_map.motor_map_plugin import MotorMapPlugin
 
     QPyDesignerCustomWidgetCollection.addCustomWidget(MotorMapPlugin())
 

bec_widgets/widgets/plots/multi_waveform/multi_waveform.py

@@ -12,13 +12,13 @@ from qtpy.QtWidgets import QWidget
 from bec_widgets.qt_utils.error_popups import SafeProperty, SafeSlot
 from bec_widgets.qt_utils.side_panel import SidePanel
 from bec_widgets.utils import Colors, ConnectionConfig
-from bec_widgets.widgets.plots_next_gen.multi_waveform.settings.control_panel import (
+from bec_widgets.widgets.plots.multi_waveform.settings.control_panel import (
     MultiWaveformControlPanel,
 )
-from bec_widgets.widgets.plots_next_gen.multi_waveform.toolbar_bundles.monitor_selection import (
+from bec_widgets.widgets.plots.multi_waveform.toolbar_bundles.monitor_selection import (
     MultiWaveformSelectionToolbarBundle,
 )
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/multi_waveform/multi_waveform_plugin.py

@@ -4,7 +4,7 @@
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 
 from bec_widgets.utils.bec_designer import designer_material_icon
-from bec_widgets.widgets.plots_next_gen.multi_waveform.multi_waveform import MultiWaveform
+from bec_widgets.widgets.plots.multi_waveform.multi_waveform import MultiWaveform
 
 DOM_XML = """
 <ui language='c++'>
@@ -27,7 +27,7 @@ class MultiWaveformPlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
         return DOM_XML
 
     def group(self):
-        return "Plot Widgets Next Gen"
+        return "Plot Widgets"
 
     def icon(self):
         return designer_material_icon(MultiWaveform.ICON_NAME)

bec_widgets/widgets/plots/multi_waveform/register_multi_waveform.py

@@ -6,9 +6,7 @@ def main():  # pragma: no cover
         return
     from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
 
-    from bec_widgets.widgets.plots_next_gen.multi_waveform.multi_waveform_plugin import (
+    from bec_widgets.widgets.plots.multi_waveform.multi_waveform_plugin import MultiWaveformPlugin
-        MultiWaveformPlugin,
-    )
 
     QPyDesignerCustomWidgetCollection.addCustomWidget(MultiWaveformPlugin())
 

bec_widgets/widgets/plots/plot_base.py

@@ -19,12 +19,12 @@ from bec_widgets.utils.fps_counter import FPSCounter
 from bec_widgets.utils.plot_indicator_items import BECArrowItem, BECTickItem
 from bec_widgets.utils.widget_state_manager import WidgetStateManager
 from bec_widgets.widgets.containers.layout_manager.layout_manager import LayoutManagerWidget
-from bec_widgets.widgets.plots_next_gen.setting_menus.axis_settings import AxisSettings
+from bec_widgets.widgets.plots.setting_menus.axis_settings import AxisSettings
-from bec_widgets.widgets.plots_next_gen.toolbar_bundles.mouse_interactions import (
+from bec_widgets.widgets.plots.toolbar_bundles.mouse_interactions import (
     MouseInteractionToolbarBundle,
 )
-from bec_widgets.widgets.plots_next_gen.toolbar_bundles.plot_export import PlotExportBundle
+from bec_widgets.widgets.plots.toolbar_bundles.plot_export import PlotExportBundle
-from bec_widgets.widgets.plots_next_gen.toolbar_bundles.roi_bundle import ROIBundle
+from bec_widgets.widgets.plots.toolbar_bundles.roi_bundle import ROIBundle
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/scatter_waveform/register_scatter_waveform.py

@@ -6,7 +6,7 @@ def main():  # pragma: no cover
         return
     from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
 
-    from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform_plugin import (
+    from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform_plugin import (
         ScatterWaveformPlugin,
     )
 

bec_widgets/widgets/plots/scatter_waveform/scatter_curve.py

@@ -11,7 +11,7 @@ from qtpy import QtCore
 from bec_widgets.utils import BECConnector, Colors, ConnectionConfig
 
 if TYPE_CHECKING:  # pragma: no cover
-    from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform import ScatterWaveform
+    from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform import ScatterWaveform
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/scatter_waveform/scatter_waveform.py

@@ -14,13 +14,13 @@ from bec_widgets.qt_utils.settings_dialog import SettingsDialog
 from bec_widgets.qt_utils.toolbar import MaterialIconAction
 from bec_widgets.utils import Colors, ConnectionConfig
 from bec_widgets.utils.colors import set_theme
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_curve import (
+from bec_widgets.widgets.plots.scatter_waveform.scatter_curve import (
     ScatterCurve,
     ScatterCurveConfig,
     ScatterDeviceSignal,
 )
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.settings.scatter_curve_setting import (
+from bec_widgets.widgets.plots.scatter_waveform.settings.scatter_curve_setting import (
     ScatterCurveSettings,
 )
 

bec_widgets/widgets/plots/scatter_waveform/scatter_waveform_plugin.py

@@ -4,7 +4,7 @@
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 
 from bec_widgets.utils.bec_designer import designer_material_icon
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform import ScatterWaveform
+from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform import ScatterWaveform
 
 DOM_XML = """
 <ui language='c++'>
@@ -27,7 +27,7 @@ class ScatterWaveformPlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
         return DOM_XML
 
     def group(self):
-        return "Plot Widgets Next Gen"
+        return "Plot Widgets"
 
     def icon(self):
         return designer_material_icon(ScatterWaveform.ICON_NAME)

bec_widgets/widgets/plots/waveform/curve.py

@@ -11,7 +11,7 @@ from qtpy import QtCore
 from bec_widgets.utils import BECConnector, Colors, ConnectionConfig
 
 if TYPE_CHECKING:  # pragma: no cover
-    from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+    from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/waveform/register_waveform.py

@@ -6,7 +6,7 @@ def main():  # pragma: no cover
         return
     from PySide6.QtDesigner import QPyDesignerCustomWidgetCollection
 
-    from bec_widgets.widgets.plots_next_gen.waveform.waveform_plugin import WaveformPlugin
+    from bec_widgets.widgets.plots.waveform.waveform_plugin import WaveformPlugin
 
     QPyDesignerCustomWidgetCollection.addCustomWidget(WaveformPlugin())
 

bec_widgets/widgets/plots/waveform/settings/curve_settings/curve_setting.py

@@ -17,10 +17,10 @@ from bec_widgets.qt_utils.settings_dialog import SettingWidget
 from bec_widgets.widgets.control.device_input.device_line_edit.device_line_edit import (
     DeviceLineEdit,
 )
-from bec_widgets.widgets.plots_next_gen.waveform.settings.curve_settings.curve_tree import CurveTree
+from bec_widgets.widgets.plots.waveform.settings.curve_settings.curve_tree import CurveTree
 
 if TYPE_CHECKING:  # pragma: no cover
-    from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+    from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 
 class CurveSetting(SettingWidget):

bec_widgets/widgets/plots/waveform/settings/curve_settings/curve_tree.py

@@ -29,11 +29,11 @@ from bec_widgets.widgets.control.device_input.device_line_edit.device_line_edit
     DeviceLineEdit,
 )
 from bec_widgets.widgets.dap.dap_combo_box.dap_combo_box import DapComboBox
-from bec_widgets.widgets.plots_next_gen.waveform.curve import CurveConfig, DeviceSignal
+from bec_widgets.widgets.plots.waveform.curve import CurveConfig, DeviceSignal
 from bec_widgets.widgets.utility.visual.colormap_widget.colormap_widget import BECColorMapWidget
 
 if TYPE_CHECKING:  # pragma: no cover
-    from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+    from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 
 class ColorButton(QPushButton):

bec_widgets/widgets/plots/waveform/waveform.py

@@ -19,12 +19,10 @@ from bec_widgets.utils import ConnectionConfig
 from bec_widgets.utils.bec_signal_proxy import BECSignalProxy
 from bec_widgets.utils.colors import Colors, set_theme
 from bec_widgets.widgets.dap.lmfit_dialog.lmfit_dialog import LMFitDialog
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
-from bec_widgets.widgets.plots_next_gen.waveform.curve import Curve, CurveConfig, DeviceSignal
+from bec_widgets.widgets.plots.waveform.curve import Curve, CurveConfig, DeviceSignal
-from bec_widgets.widgets.plots_next_gen.waveform.settings.curve_settings.curve_setting import (
+from bec_widgets.widgets.plots.waveform.settings.curve_settings.curve_setting import CurveSetting
-    CurveSetting,
+from bec_widgets.widgets.plots.waveform.utils.roi_manager import WaveformROIManager
-)
-from bec_widgets.widgets.plots_next_gen.waveform.utils.roi_manager import WaveformROIManager
 
 logger = bec_logger.logger
 

bec_widgets/widgets/plots/waveform/waveform_plugin.py

@@ -4,7 +4,7 @@
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 
 from bec_widgets.utils.bec_designer import designer_material_icon
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 DOM_XML = """
 <ui language='c++'>
@@ -27,7 +27,7 @@ class WaveformPlugin(QDesignerCustomWidgetInterface):  # pragma: no cover
         return DOM_XML
 
     def group(self):
-        return "Plot Widgets Next Gen"
+        return "Plot Widgets"
 
     def icon(self):
         return designer_material_icon(Waveform.ICON_NAME)

tests/unit_tests/test_axis_settings.py

@@ -1,8 +1,8 @@
 import pytest
 from qtpy.QtWidgets import QDoubleSpinBox, QLineEdit
 
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase
+from bec_widgets.widgets.plots.plot_base import PlotBase
-from bec_widgets.widgets.plots_next_gen.setting_menus.axis_settings import AxisSettings
+from bec_widgets.widgets.plots.setting_menus.axis_settings import AxisSettings
 from tests.unit_tests.client_mocks import mocked_client
 from tests.unit_tests.conftest import create_widget
 

tests/unit_tests/test_color_utils.py

@@ -7,7 +7,7 @@ from qtpy.QtWidgets import QVBoxLayout, QWidget
 from bec_widgets.utils import Colors, ConnectionConfig
 from bec_widgets.utils.bec_widget import BECWidget
 from bec_widgets.utils.colors import apply_theme
-from bec_widgets.widgets.plots_next_gen.waveform.curve import CurveConfig
+from bec_widgets.widgets.plots.waveform.curve import CurveConfig
 from tests.unit_tests.client_mocks import mocked_client
 from tests.unit_tests.conftest import create_widget
 

tests/unit_tests/test_curve_settings.py

@@ -4,11 +4,9 @@ from unittest.mock import MagicMock, patch
 import pytest
 from qtpy.QtWidgets import QComboBox, QVBoxLayout
 
-from bec_widgets.widgets.plots_next_gen.waveform.settings.curve_settings.curve_setting import (
+from bec_widgets.widgets.plots.waveform.settings.curve_settings.curve_setting import CurveSetting
-    CurveSetting,
+from bec_widgets.widgets.plots.waveform.settings.curve_settings.curve_tree import CurveTree
-)
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
-from bec_widgets.widgets.plots_next_gen.waveform.settings.curve_settings.curve_tree import CurveTree
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
 from tests.unit_tests.client_mocks import dap_plugin_message, mocked_client, mocked_client_with_dap
 from tests.unit_tests.conftest import create_widget
 

tests/unit_tests/test_image_view_next_gen.py

@@ -2,7 +2,7 @@ import numpy as np
 import pyqtgraph as pg
 import pytest
 
-from bec_widgets.widgets.plots_next_gen.image.image import Image
+from bec_widgets.widgets.plots.image.image import Image
 from tests.unit_tests.client_mocks import mocked_client
 from tests.unit_tests.conftest import create_widget
 

tests/unit_tests/test_motor_map_next_gen.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pyqtgraph as pg
 
-from bec_widgets.widgets.plots_next_gen.motor_map.motor_map import MotorMap
+from bec_widgets.widgets.plots.motor_map.motor_map import MotorMap
 from tests.unit_tests.client_mocks import mocked_client
 
 from .conftest import create_widget
@@ -210,7 +210,7 @@ def test_motor_map_limit_map(qtbot, mocked_client):
 
     from qtpy import QtCore
 
-    from bec_widgets.widgets.plots_next_gen.motor_map.motor_map import FilledRectItem
+    from bec_widgets.widgets.plots.motor_map.motor_map import FilledRectItem
 
     assert isinstance(limit_map, FilledRectItem)
     rect = limit_map.boundingRect()

tests/unit_tests/test_multi_waveform_next_gen.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from bec_widgets.widgets.plots_next_gen.multi_waveform.multi_waveform import MultiWaveform
+from bec_widgets.widgets.plots.multi_waveform.multi_waveform import MultiWaveform
 from tests.unit_tests.client_mocks import mocked_client
 
 from .conftest import create_widget

tests/unit_tests/test_plot_base_next_gen.py

@@ -1,4 +1,4 @@
-from bec_widgets.widgets.plots_next_gen.plot_base import PlotBase, UIMode
+from bec_widgets.widgets.plots.plot_base import PlotBase, UIMode
 
 from .client_mocks import mocked_client
 from .conftest import create_widget

tests/unit_tests/test_scatter_waveform.py

@@ -2,11 +2,11 @@ import json
 
 import numpy as np
 
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_curve import (
+from bec_widgets.widgets.plots.scatter_waveform.scatter_curve import (
     ScatterCurveConfig,
     ScatterDeviceSignal,
 )
-from bec_widgets.widgets.plots_next_gen.scatter_waveform.scatter_waveform import ScatterWaveform
+from bec_widgets.widgets.plots.scatter_waveform.scatter_waveform import ScatterWaveform
 from tests.unit_tests.client_mocks import create_dummy_scan_item, mocked_client
 
 from .conftest import create_widget

tests/unit_tests/test_utils_plot_indicators.py

@@ -1,7 +1,7 @@
 import pytest
 from qtpy.QtCore import QPointF
 
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
 
 from .client_mocks import mocked_client
 

tests/unit_tests/test_waveform_next_gen.py

@@ -6,9 +6,9 @@ import pyqtgraph as pg
 import pytest
 from pyqtgraph.graphicsItems.DateAxisItem import DateAxisItem
 
-from bec_widgets.widgets.plots_next_gen.plot_base import UIMode
+from bec_widgets.widgets.plots.plot_base import UIMode
-from bec_widgets.widgets.plots_next_gen.waveform.curve import DeviceSignal
+from bec_widgets.widgets.plots.waveform.curve import DeviceSignal
-from bec_widgets.widgets.plots_next_gen.waveform.waveform import Waveform
+from bec_widgets.widgets.plots.waveform.waveform import Waveform
 from tests.unit_tests.client_mocks import (
     DummyData,
     create_dummy_scan_item,