diff --git a/api/src/opentrons/protocol_engine/commands/absorbance_reader/close_lid.py b/api/src/opentrons/protocol_engine/commands/absorbance_reader/close_lid.py index ae3033473a5..302a192634a 100644 --- a/api/src/opentrons/protocol_engine/commands/absorbance_reader/close_lid.py +++ b/api/src/opentrons/protocol_engine/commands/absorbance_reader/close_lid.py @@ -135,8 +135,8 @@ async def execute( ) state_update = StateUpdate() - state_update.lid_status = BaseLabwareData( - id=loaded_lid.id, new_location=new_location, offset_id=new_offset_id + state_update.set_lid_status( + labware_id=loaded_lid.id, location=new_location, offset_id=new_offset_id ) return SuccessData( diff --git a/api/src/opentrons/protocol_engine/commands/absorbance_reader/open_lid.py b/api/src/opentrons/protocol_engine/commands/absorbance_reader/open_lid.py index 41d8e3091b6..b40b3bbf6f9 100644 --- a/api/src/opentrons/protocol_engine/commands/absorbance_reader/open_lid.py +++ b/api/src/opentrons/protocol_engine/commands/absorbance_reader/open_lid.py @@ -130,8 +130,8 @@ async def execute(self, params: OpenLidParams) -> SuccessData[OpenLidResult, Non labware_location=new_location, ) - state_update.lid_status = BaseLabwareData( - id=loaded_lid.id, new_location=new_location, offset_id=new_offset_id + state_update.set_lid_status( + labware_id=loaded_lid.id, location=new_location, offset_id=new_offset_id ) return SuccessData( diff --git a/api/src/opentrons/protocol_engine/state/update_types.py b/api/src/opentrons/protocol_engine/state/update_types.py index 88ad4baa94d..a152c99c58e 100644 --- a/api/src/opentrons/protocol_engine/state/update_types.py +++ b/api/src/opentrons/protocol_engine/state/update_types.py @@ -196,6 +196,16 @@ def set_reloaded_labware( id=labware_id, new_location=location, offset_id=offset_id ) + def set_lid_status( + self, + location: LabwareLocation, + labware_id: str, + offset_id: typing.Optional[str], + ) -> None: + self.lid_status = BaseLabwareData( + id=labware_id, new_location=location, offset_id=offset_id + ) + def clear_all_pipette_locations(self) -> None: """Mark all pipettes as having an unknown location.""" self.pipette_location = CLEAR