Skip to content

Commit 5f1b7ad

Browse files
committed
fix Location types
1 parent b1ac0f8 commit 5f1b7ad

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pvlib/location.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,26 +58,28 @@ def __init__(
5858
self,
5959
latitude: float,
6060
longitude: float,
61-
tz: Union[str, int, float, pytz.timezone] = 'UTC',
61+
tz: Union[str, int, float, datetime.timezone, datetime.tzinfo] = 'UTC',
6262
altitude: float = 0,
6363
name: str = None
64-
):
64+
) -> None:
6565

6666
self.latitude = latitude
6767
self.longitude = longitude
6868

69+
self.tz: Union[str, int, float]
70+
self.pytz: datetime.tzinfo # not used within pvlib
6971
if isinstance(tz, str):
7072
self.tz = tz
7173
self.pytz = pytz.timezone(tz)
7274
elif isinstance(tz, datetime.timezone):
7375
self.tz = 'UTC'
7476
self.pytz = pytz.UTC
7577
elif isinstance(tz, datetime.tzinfo):
76-
self.tz = tz.zone
78+
self.tz = str(tz)
7779
self.pytz = tz
7880
elif isinstance(tz, (int, float)):
7981
self.tz = tz
80-
self.pytz = pytz.FixedOffset(tz*60)
82+
self.pytz = pytz.FixedOffset(int(tz*60))
8183
else:
8284
raise TypeError('Invalid tz specification')
8385

0 commit comments

Comments
 (0)