diff options
Diffstat (limited to 'klippy/extras/stepper_enable.py')
| -rw-r--r-- | klippy/extras/stepper_enable.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/klippy/extras/stepper_enable.py b/klippy/extras/stepper_enable.py index cd1d34c6..e8b41c0a 100644 --- a/klippy/extras/stepper_enable.py +++ b/klippy/extras/stepper_enable.py @@ -84,10 +84,10 @@ class PrinterStepperEnable: gcode.register_command("SET_STEPPER_ENABLE", self.cmd_SET_STEPPER_ENABLE, desc=self.cmd_SET_STEPPER_ENABLE_help) - def register_stepper(self, stepper, pin): - name = stepper.get_name() - enable = setup_enable_pin(self.printer, pin) - self.enable_lines[name] = EnableTracking(stepper, enable) + def register_stepper(self, config, mcu_stepper): + name = mcu_stepper.get_name() + enable = setup_enable_pin(self.printer, config.get('enable_pin', None)) + self.enable_lines[name] = EnableTracking(mcu_stepper, enable) def motor_off(self): toolhead = self.printer.lookup_object('toolhead') toolhead.dwell(DISABLE_STALL_TIME) |