diff --git a/klippy/extras/probe.py b/klippy/extras/probe.py index 73fd8927b..c57f9ef1d 100644 --- a/klippy/extras/probe.py +++ b/klippy/extras/probe.py @@ -38,7 +38,7 @@ def calc_probe_z_average(positions, method='average'): # Helper to implement common probing commands class ProbeCommandHelper: def __init__(self, config, probe, query_endstop=None, - replace_z_offset=False): + can_set_z_offset=True): self.printer = config.get_printer() self.probe = probe self.query_endstop = query_endstop @@ -55,16 +55,16 @@ class ProbeCommandHelper: desc=self.cmd_PROBE_help) # PROBE_CALIBRATE command self.probe_calibrate_info = None - gcode.register_command('PROBE_CALIBRATE', self.cmd_PROBE_CALIBRATE, - desc=self.cmd_PROBE_CALIBRATE_help) + if can_set_z_offset: + gcode.register_command('PROBE_CALIBRATE', self.cmd_PROBE_CALIBRATE, + desc=self.cmd_PROBE_CALIBRATE_help) # Other commands gcode.register_command('PROBE_ACCURACY', self.cmd_PROBE_ACCURACY, desc=self.cmd_PROBE_ACCURACY_help) - if replace_z_offset: - return - gcode.register_command('Z_OFFSET_APPLY_PROBE', - self.cmd_Z_OFFSET_APPLY_PROBE, - desc=self.cmd_Z_OFFSET_APPLY_PROBE_help) + if can_set_z_offset: + gcode.register_command('Z_OFFSET_APPLY_PROBE', + self.cmd_Z_OFFSET_APPLY_PROBE, + desc=self.cmd_Z_OFFSET_APPLY_PROBE_help) def _move(self, coord, speed): self.printer.lookup_object('toolhead').manual_move(coord, speed) def get_status(self, eventtime): diff --git a/klippy/extras/probe_eddy_current.py b/klippy/extras/probe_eddy_current.py index 1befaf6e9..6952480a3 100644 --- a/klippy/extras/probe_eddy_current.py +++ b/klippy/extras/probe_eddy_current.py @@ -687,7 +687,7 @@ class PrinterEddyProbe: self.calibration, self.probe_offsets) # Register with main probe interface self.cmd_helper = probe.ProbeCommandHelper(config, self, - replace_z_offset=True) + can_set_z_offset=False) self.probe_session = probe.ProbeSessionHelper( config, self.param_helper, self._start_descend_wrapper) self.printer.add_object('probe', self)