[HOTFIX-#238] Fix an incompatible return type introduced in #278

raiden-network · Sep 6, 2019 · 20bce25 · 20bce25
2 parents a06a432 + 16704be
commit 20bce25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scenario_player/utils/token.py b/scenario_player/utils/token.py
@@ -1,6 +1,6 @@
 import json
 import pathlib
-from typing import Tuple, Union
+from typing import Optional, Tuple, Union
 
 import structlog
 from eth_utils import decode_hex, to_checksum_address
@@ -395,7 +395,7 @@ def mint(
             **kwargs,
         )
 
-    def update_allowance(self) -> Union[Tuple[str, int], None]:
+    def update_allowance(self) -> Tuple[Optional[str], int]:
         """Update the UD Token Contract allowance depending on the number of configured nodes.
 
         If the UD Token Contract's allowance is sufficient, this is a no-op.
@@ -414,7 +414,7 @@ def update_allowance(self) -> Union[Tuple[str, int], None]:
 
         if not udt_allowance < required_allowance:
             log.debug("UDTC allowance sufficient")
-            return
+            return None, required_allowance
 
         log.debug("UDTC allowance insufficient, updating")
         params = {