Clean up Python script

wpilibsuite · Sep 23, 2023 · 987fd80 · 987fd80
1 parent f39f09e
commit 987fd80
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/time_plots.py b/scripts/time_plots.py
@@ -53,8 +53,10 @@
         unit_columns = columns[3:]
 
     prepared_data = pd.DataFrame(columns=columns)
-    for test in (test for test in raw_data.keys() if "-" in test):
-        formatted_entry = [[pt[0]] + [test] + pt[1:] for pt in raw_data[test]]
+    for test in raw_data.keys():
+        if "-" not in test:
+            continue
+        formatted_entry = [[pt[0], test, *pt[1:]] for pt in raw_data[test]]
         prepared_data = pd.concat(
             [prepared_data, pd.DataFrame(formatted_entry, columns=columns)]
         )