Added sample heater display mode check in custom_pid function.

simonsobs · Dec 28, 2023 · 0a7134c · 0a7134c
1 parent 06ddfd4
commit 0a7134c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/socs/agents/lakeshore372/agent.py b/socs/agents/lakeshore372/agent.py
@@ -1008,6 +1008,12 @@ def custom_pid(self, session, params):
                     session.add_message('Changing control to Open Loop mode for sample PID.')
                     self.module.sample_heater.set_mode("Open Loop")
 
+                # Check we're in the correct display mode for servo.
+                # The custom PID expects that the sample heater is in 'current' mode
+                if self.module.sample_heater.display != "current":
+                    session.add_message("Changing display mode to \'current\' for sample PID.")
+                    self.module.sample_heater.set_heater_display("current")
+
             if heater == 'still':
                 # Check we're in correct control mode for servo.
                 if self.module.still_heater.mode != 'Open Loop':