mypy warnings

squishykid · Jun 20, 2023 · eda9478 · eda9478
1 parent 69faf66
commit eda9478
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/solax/discovery.py b/solax/discovery.py
@@ -1,5 +1,5 @@
 import asyncio
-from typing import Type
+from typing import Type, List
 
 from solax.http_client import all_variations
 from solax.inverter import Inverter, InverterError
@@ -14,7 +14,7 @@
     XHybrid,
 )
 
-REGISTRY: list[Type[Inverter]] = [
+REGISTRY: List[Type[Inverter]] = [
     XHybrid,
     X3,
     X3V34,
@@ -34,7 +34,7 @@ class DiscoveryError(Exception):
 
 
 async def discover(host, port, pwd="") -> Inverter:
-    failures: list = []
+    failures: List = []
     clients = all_variations(host, port, pwd)
     pending = set()
 

diff --git a/solax/inverter.py b/solax/inverter.py
@@ -15,18 +15,18 @@
 
 
 class InverterRawResponse(TypedDict):
-    Data: list[float]
+    Data: List[float]
     sn: Optional[str]
     SN: Optional[str]
     version: Optional[str]
     ver: Optional[str]
     type: Union[int, str]
-    Information: Optional[list[Any]]
+    Information: Optional[List[Any]]
 
 
 @dataclass
 class InverterResponse:
-    data: dict[str, float]
+    data: Dict[str, float]
     serial_number: str
     version: str
     type: Union[int, str]
@@ -102,7 +102,7 @@ async def get_data(self) -> InverterResponse:
 
     def map_response_v2(
         self, inverter_response: InverterRawResponse
-    ) -> dict[str, float]:
+    ) -> Dict[str, float]:
         data = inverter_response["Data"]
         highest_index = max(
             (max(v.indexes) for v in self.inverter_definition().mapping.values())