From f0dbeb0493e6866c913c099e002cf9bd50103135 Mon Sep 17 00:00:00 2001 From: Kevin O'Connor Date: Mon, 19 Jul 2021 13:22:40 -0400 Subject: stepper: Simplify helper module registration of stepper motors Signed-off-by: Kevin O'Connor --- klippy/extras/stepper_enable.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'klippy/extras/stepper_enable.py') 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) -- cgit v1.2.3-70-g09d2