From 510073d2f20cb7ce170b0d256cb5b5796806a0e7 Mon Sep 17 00:00:00 2001 From: gac-x01da Date: Thu, 1 May 2025 10:39:26 +0200 Subject: [PATCH] fix(reffoil-changer): add scaninfo to __init__ signature --- debye_bec/devices/reffoilchanger.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/debye_bec/devices/reffoilchanger.py b/debye_bec/devices/reffoilchanger.py index 9103b46..64a4a4d 100644 --- a/debye_bec/devices/reffoilchanger.py +++ b/debye_bec/devices/reffoilchanger.py @@ -7,6 +7,11 @@ from ophyd.status import DeviceStatus from ophyd_devices.interfaces.base_classes.psi_device_base import PSIDeviceBase from ophyd_devices.utils.errors import DeviceStopError +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from bec_lib.devicemanager import ScanInfo + class Status(int, enum.Enum): """Enum class for the status field""" @@ -90,8 +95,8 @@ class Reffoilchanger(PSIDeviceBase): foil37 = Cpt(EpicsSignalRO, suffix="ES-REFFOIL:FOIL37.DESC", kind="config", doc='Foil 37') foil38 = Cpt(EpicsSignalRO, suffix="ES-REFFOIL:FOIL38.DESC", kind="config", doc='Foil 38') - def __init__(self, name:str, prefix, **kwargs): - super().__init__(name=name, prefix=prefix, **kwargs) + def __init__(self, *, name: str, prefix: str = "", scan_info: ScanInfo | None = None, **kwargs) + super().__init__(name=name, prefix=prefix, scan_info=scan_info, **kwargs) self.foils = [ self.foil01, self.foil02, self.foil03, self.foil04, self.foil05, self.foil06,