Help the typechecker some more

Author: mfisher87

antarctica_today/compute_mean_climatology.py

@@ -429,7 +429,8 @@ def create_partial_year_melt_anomaly_tif(
 
     # Round to integers, if it isn't already.
     anomalies_int: numpy.ndarray = numpy.array(
-        numpy.round(anomaly_this_season_so_far), dtype=numpy.int32
+        numpy.round(anomaly_this_season_so_far),
+        dtype=numpy.int32,
     )
 
     # If dest_fname is None, create it.
@@ -591,9 +592,7 @@ def create_annual_melt_sum_tif(
     dt_list = list(datetimes_dict.keys())
 
     if year == "all":
-        years = numpy.unique([dt.year for dt in dt_list])
-        years.sort()
-
+        years = sorted({dt.year for dt in dt_list})
     else:
         assert year == int(year)
         years = [year]

antarctica_today/melt_array_picklefile.py

@@ -15,7 +15,7 @@
 import pickle
 import re
 from pathlib import Path
-from typing import Any
+from typing import Any, Dict, Tuple
 
 import numpy
 from osgeo import gdal
@@ -323,15 +323,15 @@ def _filter_out_erroneous_swaths(model_array, datetimes_dict):
 
 
 def read_gap_filled_melt_picklefile(
-    picklefile=gap_filled_melt_picklefile, verbose=True
-):
+    picklefile: Path = gap_filled_melt_picklefile,
+    verbose: bool = True,
+) -> Tuple[numpy.ndarray, Dict[datetime.datetime, int]]:
     """Read the gap-filled picklefile, return to user."""
     if verbose:
         print("Reading", picklefile)
 
-    f = open(picklefile, "rb")
-    array, dt_dict = pickle.load(f)
-    f.close()
+    with open(picklefile, "rb") as f:
+        array, dt_dict = pickle.load(f)
 
     return array, dt_dict