From b593d01ec9f7164321f6ba6f3afd21abbfc9abe2 Mon Sep 17 00:00:00 2001 From: Nikhil Parasaram Date: Wed, 27 Nov 2019 16:39:38 +0000 Subject: [PATCH] Make some small changes --- mythril/laser/ethereum/state/world_state.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/mythril/laser/ethereum/state/world_state.py b/mythril/laser/ethereum/state/world_state.py index 51185ec6..8a68fcbf 100644 --- a/mythril/laser/ethereum/state/world_state.py +++ b/mythril/laser/ethereum/state/world_state.py @@ -9,6 +9,9 @@ from ethereum.utils import mk_contract_address from mythril.laser.ethereum.state.account import Account from mythril.laser.ethereum.state.annotation import StateAnnotation from mythril.laser.ethereum.state.constraints import Constraints +from mythril.laser.ethereum.plugins.implementations.plugin_annotations import ( + WSDependencyAnnotation, +) if TYPE_CHECKING: from mythril.laser.ethereum.cfg import Node @@ -38,7 +41,7 @@ class WorldState: self.node = None # type: Optional['Node'] self.transaction_sequence = transaction_sequence or [] - self._annotations = annotations or [] + self._annotations = annotations or [] # type: List[WSDependencyAnnotation] def merge_states(self, state: "WorldState"): """ @@ -269,7 +272,7 @@ class WorldState: new_account.storage = storage self.put_account(new_account) - def annotate(self, annotation: StateAnnotation) -> None: + def annotate(self, annotation: WSDependencyAnnotation) -> None: """ :param annotation: @@ -277,7 +280,7 @@ class WorldState: self._annotations.append(annotation) @property - def annotations(self) -> List[StateAnnotation]: + def annotations(self) -> List[WSDependencyAnnotation]: """ :return: