From a52d73175894afa6a55f557a3e94c34bf06c8908 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Beaucamp?=
 <88246672+fbeauKmi@users.noreply.github.com>
Date: Fri, 17 Nov 2023 18:27:11 +0100
Subject: [PATCH] Update heaters.py (#98)

Allow PID_V parameter to be modify whithout reloading too
---
 klippy/extras/heaters.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/klippy/extras/heaters.py b/klippy/extras/heaters.py
index 7d8c9bef6..243031985 100644
--- a/klippy/extras/heaters.py
+++ b/klippy/extras/heaters.py
@@ -194,8 +194,8 @@ def cmd_SET_HEATER_TEMPERATURE(self, gcmd):
     cmd_SET_HEATER_PID_help = "Sets a heater PID parameter"
 
     def cmd_SET_HEATER_PID(self, gcmd):
-        if not isinstance(self.control, ControlPID):
-            raise gcmd.error("Not a PID controlled heater")
+        if not isinstance(self.control, (ControlPID, ControlVelocityPID)):
+            raise gcmd.error("Not a PID/PID_V controlled heater")
         kp = gcmd.get_float("KP", None)
         if kp is not None:
             self.control.Kp = kp / PID_PARAM_BASE