tidied up speed test plot.

cylammarco · Apr 25, 2023 · e4be114 · e4be114
1 parent 871691c
commit e4be114
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/speed_test/plot_speed_test_results.py b/speed_test/plot_speed_test_results.py
@@ -30,6 +30,7 @@
     axs[0].plot(
         insizes,
         p / n,
+        label=f"numba with output size: {size}",
         color=red_gradient[j],
     )
 
@@ -53,7 +54,6 @@
     axs[1].plot(
         insizes,
         p_e / n_e,
-        label=f"numba with output size: {size}",
         color=red_gradient[k],
     )
 
@@ -74,7 +74,6 @@
 axs[1].grid()
 
 axs[0].legend()
-axs[1].legend()
 
 axs[0].xaxis.set_major_formatter(ScalarFormatter(useMathText=False))
 axs[1].xaxis.set_major_formatter(ScalarFormatter(useMathText=False))
@@ -90,8 +89,8 @@
 axs[0].set_xlim(150, 4250)
 axs[1].set_xlim(150, 4250)
 
-axs[0].set_ylim(0, 250)
-axs[1].set_ylim(0, 250)
+axs[0].set_ylim(0, 400)
+axs[1].set_ylim(0, 400)
 
 axs[1].set_yticklabels([])
 

diff --git a/speed_test/speed_test.png b/speed_test/speed_test.png