fix res_tester missing

Frix-x · Dec 16, 2024 · 6376c84 · 6376c84
1 parent f8f3f16
commit 6376c84
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/shaketune/commands/axes_shaper_calibration.py b/shaketune/commands/axes_shaper_calibration.py
@@ -119,7 +119,7 @@ def axes_shaper_calibration(gcmd, config, st_process: ShakeTuneProcess) -> None:
         # Then do the actual measurements
         ConsoleOutput.print(f'Measuring {config["label"]}...')
         accelerometer.start_recording(measurements_manager, name=config['label'], append_time=True)
-        vibrate_axis(toolhead, gcode, config['direction'], min_freq, max_freq, hz_per_sec, accel_per_hz)
+        vibrate_axis(toolhead, gcode, config['direction'], min_freq, max_freq, hz_per_sec, accel_per_hz, res_tester)
         accelerometer.stop_recording()
         accelerometer.wait_for_samples()
         toolhead.dwell(0.5)

diff --git a/shaketune/commands/compare_belts_responses.py b/shaketune/commands/compare_belts_responses.py
@@ -122,7 +122,7 @@ def compare_belts_responses(gcmd, config, st_process: ShakeTuneProcess) -> None:
     for config in filtered_config:
         ConsoleOutput.print(f'Measuring {config["label"]}...')
         accelerometer.start_recording(measurements_manager, name=config['label'], append_time=True)
-        vibrate_axis(toolhead, gcode, config['direction'], min_freq, max_freq, hz_per_sec, accel_per_hz)
+        vibrate_axis(toolhead, gcode, config['direction'], min_freq, max_freq, hz_per_sec, accel_per_hz, res_tester)
         accelerometer.stop_recording()
         accelerometer.wait_for_samples()
         toolhead.dwell(0.5)

diff --git a/shaketune/helpers/resonance_test.py b/shaketune/helpers/resonance_test.py
@@ -307,7 +307,7 @@ def _project_distance(distance, normalized_direction):
         )
 
 
-def vibrate_axis(toolhead, gcode, axis_direction, min_freq, max_freq, hz_per_sec, accel_per_hz, res_tester=None):
+def vibrate_axis(toolhead, gcode, axis_direction, min_freq, max_freq, hz_per_sec, accel_per_hz, res_tester):
     manager = ResonanceTestManager(toolhead, gcode, res_tester)
     manager.vibrate_axis(axis_direction, min_freq, max_freq, hz_per_sec, accel_per_hz)