setter instead of direct set

Opentrons · Sep 20, 2024 · d4c06ce · d4c06ce
1 parent 880973f
commit d4c06ce
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
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
@@ -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
@@ -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
@@ -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