actually dispatch a reload command

Opentrons · Apr 19, 2024 · ba78b4d · ba78b4d
1 parent 59658e0
commit ba78b4d
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/api/src/opentrons/protocol_api/core/engine/labware.py b/api/src/opentrons/protocol_api/core/engine/labware.py
@@ -117,6 +117,13 @@ def set_calibration(self, delta: Point) -> None:
             vector=LabwareOffsetVector(x=delta.x, y=delta.y, z=delta.z),
         )
         self._engine_client.add_labware_offset(request)
+        self._engine_client.reload_labware(
+            labware_id=self._labware_id,
+            load_name=self._definition.parameters.loadName,
+            namespace=self._definition.namespace,
+            version=self._definition.version,
+            display_name=self.get_display_name(),
+        )
 
     def get_calibrated_offset(self) -> Point:
         return self._engine_client.state.geometry.get_labware_position(self._labware_id)

diff --git a/api/tests/opentrons/protocol_api/core/engine/test_labware_core.py b/api/tests/opentrons/protocol_api/core/engine/test_labware_core.py
@@ -79,6 +79,7 @@ def test_get_load_params(subject: LabwareCore) -> None:
             version=42,
             parameters=LabwareDefinitionParameters.construct(loadName="world"),  # type: ignore[call-arg]
             ordering=[],
+            metadata=LabwareDefinitionMetadata.construct(displayName="what a cool labware"),  # type: ignore[call-arg]
         )
     ],
 )
@@ -98,6 +99,9 @@ def test_set_calibration_succeeds_in_ok_location(
             version=labware_definition.version,
         )
     )
+    decoy.when(
+        mock_engine_client.state.labware.get_display_name("cool-labware")
+    ).then_return("what a cool labware")
     location = LabwareOffsetLocation(slotName=DeckSlotName.SLOT_C2)
     decoy.when(
         mock_engine_client.state.geometry.get_offset_location("cool-labware")
@@ -110,7 +114,14 @@ def test_set_calibration_succeeds_in_ok_location(
                 location=location,
                 vector=LabwareOffsetVector(x=1, y=2, z=3),
             )
-        )
+        ),
+        mock_engine_client.reload_labware(
+            labware_id="cool-labware",
+            load_name=labware_definition.parameters.loadName,
+            namespace=labware_definition.namespace,
+            version=labware_definition.version,
+            display_name="what a cool labware",
+        ),
     )