Skip to content

Commit 4aa09cb

Browse files
DominicOramtpoliaw
authored andcommitted
Fix device heirachy
1 parent d4db2aa commit 4aa09cb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/dodal/beamlines/i03.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -254,27 +254,27 @@ def synchrotron() -> Synchrotron:
254254

255255

256256
@devices.factory()
257-
def undulator(daq_configuration_path: str) -> UndulatorInKeV:
257+
def undulator(baton: Baton, daq_configuration_path: str) -> UndulatorInKeV:
258258
"""Get the i03 undulator device, instantiate it if it hasn't already been.
259259
If this is called when already instantiated in i03, it will return the existing object.
260260
"""
261261
return UndulatorInKeV(
262262
f"{BeamlinePrefix(BL).insertion_prefix}-MO-SERVC-01:",
263263
id_gap_lookup_table_path=f"{daq_configuration_path}/lookup/BeamLine_Undulator_toGap.txt",
264-
baton=baton(),
264+
baton=baton,
265265
)
266266

267267

268268
@devices.factory()
269269
def undulator_dcm(
270-
undulator: UndulatorInKeV, daq_configuration_path: str
270+
undulator: UndulatorInKeV, dcm: DCM, daq_configuration_path: str
271271
) -> UndulatorDCM:
272272
"""Get the i03 undulator DCM device, instantiate it if it hasn't already been.
273273
If this is called when already instantiated in i03, it will return the existing object.
274274
"""
275275
return UndulatorDCM(
276276
undulator=undulator,
277-
dcm=dcm(),
277+
dcm=dcm,
278278
daq_configuration_path=daq_configuration_path,
279279
)
280280

0 commit comments

Comments
 (0)