bed_mesh: Allow single probe_count

HitLuca · Jul 22, 2022 · 179eb8b · 179eb8b
1 parent be29ad2
commit 179eb8b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/optional/bed_mesh.cfg b/optional/bed_mesh.cfg
@@ -42,10 +42,10 @@ gcode:
       {% set mesh_max_y = safe_max_y %}
     {% endif %}
 
-    {% set max_x_probes = (params.PROBE_COUNT |
-                           default(bed_mesh.probe_count)).split(",")[0]|int %}
-    {% set max_y_probes = (params.PROBE_COUNT |
-                           default(bed_mesh.probe_count)).split(",")[1]|int %}
+    {% set probe_count = (params.PROBE_COUNT |
+                          default(bed_mesh.probe_count)).split(",") %}
+    {% set max_x_probes = probe_count[0]|int %}
+    {% set max_y_probes = probe_count[1]|default(max_x_probes)|int %}
 
     {% set x_probes = (max_x_probes * (mesh_max_x - mesh_min_x) /
                       (safe_max_x - safe_min_x) * probe_count_scale)