Fix missing import in run_snow_annual_map.py