Change plot ordering so colors are consistent

QuState · Feb 3, 2024 · 0b6adc7 · 0b6adc7
1 parent 4a50376
commit 0b6adc7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benches/benchmark_plots.py b/benches/benchmark_plots.py
@@ -39,7 +39,7 @@ def build_and_clean_data(
     return data
 
 
-def plot_lines(data: dict[str, list], n_range: range) -> None:
+def plot(data: dict[str, list], n_range: range) -> None:
     index = [bytes2human(2**n * (128 / 8)) for n in n_range]
     plt.figure()
 
@@ -52,9 +52,9 @@ def plot_lines(data: dict[str, list], n_range: range) -> None:
 
     df = pd.DataFrame(
         {
+            "RustFFT": np.ones(len(index)),
             "FFTW3": y0,
             "PhastFT": y1,
-            "RustFFT": np.ones(len(index)),
         },
         index=index,
     )
@@ -83,7 +83,7 @@ def main():
     assert (
         len(all_data["rustfft"]) == len(all_data["fftw3"]) == len(all_data["phastft"])
     )
-    plot_lines(all_data, n_range)
+    plot(all_data, n_range)
 
 
 if __name__ == "__main__":