lift after done

KalicoCrew · Jul 17, 2024 · b851688 · b851688
1 parent 6f49a66
commit b851688
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/klippy/extras/probe.py b/klippy/extras/probe.py
@@ -538,12 +538,13 @@ def _lift_toolhead(self):
     def _move_next(self):
         toolhead = self.printer.lookup_object("toolhead")
         # Check if done probing
+        done = False
         if len(self.results) >= len(self.probe_points):
             toolhead.get_last_move_time()
             res = self.finalize_callback(self.probe_offsets, self.results)
             if isinstance(res, (int, float)):
                 if res == 0:
-                    return True
+                    done = True
                 if self.adaptive_horizontal_move_z:
                     # then res is error
                     error = math.ceil(res)
@@ -552,9 +553,11 @@ def _move_next(self):
                         self.min_horizontal_move_z,
                     )
             elif res != "retry":
-                return True
+                done = True
             self.results = []
         self._lift_toolhead()
+        if done:
+            return True
         # Move to next XY probe point
         nextpos = list(self.probe_points[len(self.results)])
         if self.use_offsets: