Update use of leaves as iterator

Signed-off-by: Peter Schwarz <pschwarz@bitwise.io>
hyperledger-archives · May 1, 2018 · 75e1046 · 75e1046
1 parent 7b40f34
commit 75e1046
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/common/sawtooth_poet_common/validator_registry_view/validator_registry_view.py b/common/sawtooth_poet_common/validator_registry_view/validator_registry_view.py
@@ -51,7 +51,7 @@ def get_validators(self):
         leaves = self._state_view.leaves(_NAMESPACE)
         infos = [
             ValidatorRegistryView._parse_validator_info(state_data)
-            for address, state_data in leaves.items()
+            for address, state_data in leaves
             if address != validator_map_addr
         ]
         return {info.id: info for info in infos}

diff --git a/common/tests/test_validator_registry_view/mocks.py b/common/tests/test_validator_registry_view/mocks.py
@@ -36,8 +36,8 @@ def addresses(self):
 
     def leaves(self, prefix):
         """See sawtooth_validator.state.state_view.StateView.leaves"""
-        return {
+        return ({
             address: data
             for address, data in self._state.items()
             if address.startswith(prefix)
-        }
+        }).items()