bugfix/z_tilt: check if retry_result is a number before converting

KalicoCrew · Aug 12, 2024 · 8bff020 · 8bff020
1 parent 354c759
commit 8bff020
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/klippy/extras/z_tilt.py b/klippy/extras/z_tilt.py
@@ -83,7 +83,10 @@ def __init__(self, printer):
         )
 
     def check_retry_result(self, retry_result):
-        if (retry_result and retry_result.isdigit() and int(retry_result) == 0) or retry_result == "done":
+        if retry_result and (
+            (isinstance(retry_result, str) and retry_result == "done")
+            or (isinstance(retry_result, float) and int(retry_result) == 0)
+        ):
             self.applied = True
         return retry_result
 

diff --git a/klippy/extras/z_tilt_ng.py b/klippy/extras/z_tilt_ng.py
@@ -105,7 +105,10 @@ def __init__(self, printer):
         )
 
     def check_retry_result(self, retry_result):
-        if (retry_result and retry_result.isdigit() and int(retry_result) == 0) or retry_result == "done":
+        if retry_result and (
+            (isinstance(retry_result, str) and retry_result == "done")
+            or (isinstance(retry_result, float) and int(retry_result) == 0)
+        ):
             self.applied = True
         return retry_result