Change wait_steps_end function to prevent timeout error

microsoft · Jul 24, 2019 · 2decce7 · 2decce7
1 parent ab8db67
commit 2decce7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/qcodes/instrument_drivers/Attocube/Attocube_ANC300.py b/qcodes/instrument_drivers/Attocube/Attocube_ANC300.py
@@ -227,8 +227,13 @@ def ans_parser(name, ans, unit=None, parser=str):
             self.add_function('stop',
                             call_cmd='stop {}'.format(self.aid))
 
-            self.add_function('wait_steps_end',
-                            call_cmd='stepw {}'.format(self.aid))
+            def wait_steps_end(self):
+                old_timeout = self.root_instrument.visa_handle.timeout
+                try:
+                    self.root_instrument.visa_handle.timeout = 3600
+                    self.write_raw(f'stepw {self.aid}')
+                finally:
+                    self.root_instrument.visa_handle.timeout = old_timeout
 
             # TODO(Thibaud Ruelle): test for range before adding param
             self.add_parameter('trigger_up_pin',