diff --git a/klippy/extras/load_cell_probe.py b/klippy/extras/load_cell_probe.py index 1e08090c2..df12c9fc3 100644 --- a/klippy/extras/load_cell_probe.py +++ b/klippy/extras/load_cell_probe.py @@ -544,9 +544,11 @@ class TappingMove: # ProbeSession that implements Tap logic class TapSession: - def __init__(self, config, tapping_move, probe_params_helper): + def __init__(self, config, tapping_move, + probe_offsets, probe_params_helper): self._printer = config.get_printer() self._tapping_move = tapping_move + self._probe_offsets = probe_offsets self._probe_params_helper = probe_params_helper # Session state self._results = [] @@ -632,7 +634,8 @@ class LoadCellPrinterProbe: continuous_tare_filter_helper, config_helper) self._tapping_move = TappingMove(config, load_cell_probing_move, config_helper) - tap_session = TapSession(config, self._tapping_move, self._param_helper) + tap_session = TapSession(config, self._tapping_move, + self._probe_offsets, self._param_helper) self._probe_session = probe.ProbeSessionHelper(config, self._param_helper, tap_session.start_probe_session) # printer integration