fix types lint

Author: Xiao Li

python/setup.py

@@ -76,7 +76,7 @@ def run(self):
 setup(
     name="pylibra-beta",
     # change to 0.1.YYYYMMDDNN on release
-    version="0.5.master",
+    version="0.5.2020091001",
     description="",
     python_requires=">=3.7",
     packages=find_packages("src"),

python/src/pylibra/_jsonrpc_transport.py

@@ -44,7 +44,7 @@ class GetAccountStateResp:
     balances: typing.List[typing.Dict]
     sent_events_key: str
     received_events_key: str
-    role: typing.Union[str, typing.Dict]
+    role: typing.Dict[str, typing.Any]
     is_frozen: bool
 
 
@@ -160,7 +160,7 @@ def metadata(self) -> bytes:
         return bytes.fromhex(self._transaction["script"]["metadata"]) if self.is_p2p else b""
 
     @property
-    def vm_status(self) -> int:
+    def vm_status(self) -> typing.Dict[str, typing.Any]:
         return self._result["vm_status"]
 
 
@@ -178,15 +178,11 @@ def __init__(self, address_hex: str, state: GetAccountStateResp):
         for balance_dict in self._state.balances:
             self._balances[balance_dict["currency"]] = balance_dict["amount"]
 
-        if type(self._state.role) == str:
-            # For strings returned like "empty", "unknown"
-            self._role = typing.cast(str, self._state.role)
-        else:
-            role_dict = typing.cast(typing.Dict, self._state.role)
-            if self._state.role["type"] == "parent_vasp":
-                self._role = ParentVASP(**(role_dict))
-            elif self._state.role["type"] == "child_vasp":
-                self._role = ChildVASP(**(role_dict))
+        role_dict = typing.cast(typing.Dict, self._state.role)
+        if self._state.role["type"] == "parent_vasp":
+            self._role = ParentVASP(**(role_dict))
+        elif self._state.role["type"] == "child_vasp":
+            self._role = ChildVASP(**(role_dict))
 
     @property
     def address(self) -> bytes:

python/src/pylibra/_types.py

@@ -178,7 +178,7 @@ def metadata(self) -> bytes:
 
     @property
     @abstractmethod
-    def vm_status(self) -> int:
+    def vm_status(self) -> typing.Dict[str, typing.Any]:
         raise NotImplementedError()
 
 

python/tests/test_lcs.py

@@ -14,7 +14,10 @@ def test_lcs_e2e() -> None:
     print("Testing serialization: ")
 
     obj = libra.Transaction__GenesisTransaction(
-        libra.WriteSetPayload__Direct(libra.ChangeSet(libra.WriteSet(libra.WriteSetMut(write_set=[])), []))
+        value=libra.WriteSetPayload__Direct(
+            value=libra.ChangeSet(
+                write_set=libra.WriteSet(
+                    value=libra.WriteSetMut(write_set=[])), events=[]))
     )
     content = lcs.serialize(obj, libra.Transaction)
 
@@ -68,7 +71,7 @@ def test_stdlib() -> None:
     assert isinstance(raw_txn.payload, libra.TransactionPayload__Script)
 
     token = libra.TypeTag__Struct(
-        libra.StructTag(
+        value=libra.StructTag(
             address=make_address(b"\x00" * 15 + b"\x01"),
             module=libra.Identifier("LBR"),
             name=libra.Identifier("LBR"),