Skip to content

Commit 62ed2da

Browse files
DominicOramrtuck99
authored andcommitted
Fix device heirachy
1 parent e4740e6 commit 62ed2da

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
@@ -256,27 +256,27 @@ def synchrotron() -> Synchrotron:
256256

257257

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

269269

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

0 commit comments

Comments
 (0)