small bug fix batch_local_velocity, removed extra line in single_select

M-Sabrina · May 27, 2022 · fc405a2 · fc405a2
1 parent 6f9f98d
commit fc405a2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/min_analysis_scripts/batch_local_velocity.py b/min_analysis_scripts/batch_local_velocity.py
@@ -114,10 +114,10 @@
     with open(csv_file, "a") as csv_f:  # will append to existing
         # create the csv writer
         writer = csv.writer(csv_f, delimiter="\t")
-        writer.writerow(header)
         for n in range(velocities.size):
             writer.writerow(
                 [
+                    stackname,
                     velocities[n] * factor,
                     forward_wavevector_x[n],
                     forward_wavevector_y[n],

diff --git a/min_analysis_scripts/single_select.py b/min_analysis_scripts/single_select.py
@@ -337,7 +337,6 @@ class Action(IntEnum):
 
     savename = outpath / f"{stackname}_velocity_wheel.png"
     fig.set_size_inches(fig_size)
-    ax_sum.set_ylim((0, 1000))
     fig.tight_layout()
     fig.savefig(savename, dpi=500)
     plt.close(fig)