aboutsummaryrefslogtreecommitdiffstats
path: root/klippy/extras
diff options
context:
space:
mode:
Diffstat (limited to 'klippy/extras')
-rw-r--r--klippy/extras/heater_bed.py4
-rw-r--r--klippy/extras/pid_calibrate.py2
2 files changed, 3 insertions, 3 deletions
diff --git a/klippy/extras/heater_bed.py b/klippy/extras/heater_bed.py
index c8f68f2e..c3505528 100644
--- a/klippy/extras/heater_bed.py
+++ b/klippy/extras/heater_bed.py
@@ -19,10 +19,10 @@ class PrinterHeaterBed:
# Set Bed Temperature
gcode = self.printer.lookup_object('gcode')
temp = gcode.get_float('S', params, 0.)
- toolhead = self.printer.lookup_object('toolhead')
self.heater.set_temp(temp)
if wait and temp:
- gcode.wait_for_temperature(self.heater)
+ pheater = self.printer.lookup_object('heater')
+ pheater.wait_for_temperature(self.heater)
def cmd_M190(self, params):
# Set Bed Temperature and Wait
self.cmd_M140(params, wait=True)
diff --git a/klippy/extras/pid_calibrate.py b/klippy/extras/pid_calibrate.py
index 689084e9..49ae3bb4 100644
--- a/klippy/extras/pid_calibrate.py
+++ b/klippy/extras/pid_calibrate.py
@@ -31,7 +31,7 @@ class PIDCalibrate:
except self.printer.command_error as e:
heater.set_control(old_control)
raise
- self.gcode.wait_for_temperature(heater)
+ pheater.wait_for_temperature(heater)
heater.set_control(old_control)
if write_file:
calibrate.write_file('/tmp/heattest.txt')