Resolve typing ImportError in policy

pull/444/head
Kieran Prasch 2018-09-21 12:01:50 -07:00
parent 4e1fe0e659
commit b137ebbc76
2 changed files with 5 additions and 3 deletions

View File

@ -114,7 +114,7 @@ class BlockchainPolicy(Policy):
def __find_ursulas(self, ether_addresses: List[str], target_quantity: int, timeout: int = 120): def __find_ursulas(self, ether_addresses: List[str], target_quantity: int, timeout: int = 120):
start_time = maya.now() # Marker for timeout calculation start_time = maya.now() # Marker for timeout calculation
found_ursulas, unknown_addresses = set(), deque() # type: Tuple[set, deque] found_ursulas, unknown_addresses = set(), deque() # type: set, deque
while len(found_ursulas) < target_quantity: while len(found_ursulas) < target_quantity:
# Check for a timeout # Check for a timeout

View File

@ -1,5 +1,5 @@
from collections import OrderedDict from collections import OrderedDict
from typing import List from typing import List, Set
import maya import maya
@ -34,7 +34,9 @@ class MockPolicy(Policy):
hrac=self.hrac(), hrac=self.hrac(),
expiration=expiration) expiration=expiration)
self.consider_arrangement(network_middleware=network_middleware, arrangement=arrangement) self.consider_arrangement(network_middleware=network_middleware,
ursula=ursula,
arrangement=arrangement)
class MockPolicyCreation: class MockPolicyCreation: