Add function annotations

Inst_eddies/Analysis/plot_eddy_contours.py

@@ -18,6 +18,7 @@ mn for a 180° by 180° window.
 
 import shapefile
 import numpy as np
+from cartopy.mpl import geoaxes
 import cartopy.crs as ccrs
 from os import path
 from matplotlib import patches
@@ -25,7 +26,8 @@ import f90nml
 import sys
 import util_eddies
 
-def select_ishapes(d, SHPC, i_slice, orientation, window = None):
+def select_ishapes(d: int, SHPC: util_eddies.SHPC_class, i_slice: int,
+                   orientation: str, window: list = None):
     """Select ishapes at date d and in window. We assume date d is in
     slice i_slice of the SHPC.
 
@@ -47,8 +49,9 @@ def select_ishapes(d, SHPC, i_slice, orientation, window = None):
 
     return ishape_list_filt
 
-def snapshot(ax, ishape_list, SHPC, i_slice, orientation, *, dashed = False,
-             light = False, src_crs = ccrs.PlateCarree()):
+def snapshot(ax: geoaxes.GeoAxesSubplot, ishape_list, SHPC, i_slice,
+             orientation, *, dashed = False, light = False,
+             src_crs = ccrs.PlateCarree()):
     """Plots extrema, outermost contours and max-speed contours from a
     given SHPC, for slice i_slice, for a given orientation, for
     ishapes given by ishape_list.