From 030ae05889f440e366ed1eeca8dc749b2b74af64 Mon Sep 17 00:00:00 2001
From: Athanasius <github@miggy.org>
Date: Fri, 18 Sep 2020 12:58:16 +0100
Subject: [PATCH] Adjust return type of companion.session.station()
Now we're correctly always returning `data` it can't be None.
---
companion.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/companion.py b/companion.py
index 9176fc2e..3be504f4 100644
--- a/companion.py
+++ b/companion.py
@@ -530,7 +530,7 @@ class Session(object):
"""Perform general CAPI /profile endpoint query."""
return self.query(URL_QUERY)
- def station(self) -> Union[CAPIData, None]:
+ def station(self) -> CAPIData:
"""Perform CAPI /profile endpoint query for station data."""
data = self.query(URL_QUERY)
if not data['commander'].get('docked'):