fix data collector for grbl

misko · Mar 23, 2024 · 563484c · 563484c
1 parent 9a0160a
commit 563484c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/spf/data_collector.py b/spf/data_collector.py
@@ -495,8 +495,11 @@ def radios_to_online(self):
 
 class GrblDataCollector(DataCollector):
     def __init__(self, *args, **kwargs):
-        super(GrblDataCollector, self).__init__(*args, **kwargs)
-        self.column_names = v2_column_names(nthetas=self.yaml_config["n-thetas"])
+        super(GrblDataCollector, self).__init__(
+            *args,
+            column_names=v2_column_names(nthetas=kwargs["yaml_config"]["n-thetas"]),
+            **kwargs,
+        )
 
     def write_to_record_matrix(self, thread_idx, record_idx, data):
         tx_pos = self.position_controller.controller.position["xy"][