From a637c2f11019faecb4a4d244ea0ce9815a784b43 Mon Sep 17 00:00:00 2001
From: Kevin O'Connor <kevin@koconnor.net>
Date: Tue, 22 Dec 2020 15:22:13 -0500
Subject: [PATCH] heaters: Fix TEMPERATURE_WAIT command with extruder and
heater_bed
Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
---
klippy/extras/heaters.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/klippy/extras/heaters.py b/klippy/extras/heaters.py
index 840bbfd58..b9878e2a3 100644
--- a/klippy/extras/heaters.py
+++ b/klippy/extras/heaters.py
@@ -332,7 +332,10 @@ class PrinterHeaters:
min_temp = gcmd.get_float('MINIMUM')
if self.printer.get_start_args().get('debugoutput') is not None:
return
- sensor = self.printer.lookup_object(sensor_name)
+ if sensor_name in self.heaters:
+ sensor = self.heaters[sensor_name]
+ else:
+ sensor = self.printer.lookup_object(sensor_name)
toolhead = self.printer.lookup_object("toolhead")
reactor = self.printer.get_reactor()
eventtime = reactor.monotonic()