type annotations for Location

Author: wholmgren

pvlib/location.py

@@ -5,13 +5,14 @@
 # Will Holmgren, University of Arizona, 2014-2016.
 
 import datetime
-import warnings
+from typing import Union
 
 import pandas as pd
 import pytz
 
 from pvlib import solarposition, clearsky, atmosphere, irradiance
 
+
 class Location:
     """
     Location objects are convenient containers for latitude, longitude,
@@ -53,7 +54,14 @@ class Location:
     pvlib.pvsystem.PVSystem
     """
 
-    def __init__(self, latitude, longitude, tz='UTC', altitude=0, name=None):
+    def __init__(
+        self,
+        latitude: float,
+        longitude: float,
+        tz: Union[str, int, float, pytz.timezone] = 'UTC',
+        altitude: float = 0,
+        name: str = None
+    ):
 
         self.latitude = latitude
         self.longitude = longitude