Use OO-style

Inst_eddies/Analysis/plot_velocity.py

@@ -75,14 +75,14 @@ with netCDF4.Dataset(args.input_file) as f:
                                   vgos[lat_mask][:, lon_mask],
                                   scale = args.scale, scale_units = "width",
                                   transform = src_crs)
-        plt.quiverkey(quiver_return, 0.9, 0.9, 1, r"1 m s$^{-1}$",
-                      coordinates = "figure")
+        ax.quiverkey(quiver_return, 0.9, 0.9, 1, r"1 m s$^{-1}$",
+                     coordinates = "figure")
 
 ax.gridlines(draw_labels = True)
 ax.coastlines()
 
 if args.save:
-    plt.savefig(f"plot_velocity.{args.save}")
+    fig.savefig(f"plot_velocity.{args.save}")
     print(f'Created "plot_velocity.{args.save}".')
 else:
     plt.show()