Fix some mypy errors

This commit is contained in:
Hsiao-Wei Wang 2019-08-24 03:03:00 +08:00
parent bcdbf7dfc7
commit 17043891ff
No known key found for this signature in database
GPG Key ID: 95B070122902DEA4
2 changed files with 9 additions and 6 deletions

View File

@ -76,7 +76,7 @@ def unpack_compact_validator(compact_validator: int) -> Tuple[int, bool, int]:
""" """
Returns validator index, slashed, balance // EFFECTIVE_BALANCE_INCREMENT Returns validator index, slashed, balance // EFFECTIVE_BALANCE_INCREMENT
""" """
return compact_validator >> 16, (compact_validator >> 15) % 2, compact_validator & (2**15 - 1) return compact_validator >> 16, bool((compact_validator >> 15) % 2), compact_validator & (2**15 - 1)
``` ```
#### `committee_to_compact_committee` #### `committee_to_compact_committee`
@ -129,7 +129,7 @@ def compute_historical_state_generalized_index(earlier: ShardSlot, later: ShardS
for i in range(63, -1, -1): for i in range(63, -1, -1):
if (later - 1) & 2**i > (earlier - 1) & 2**i: if (later - 1) & 2**i > (earlier - 1) & 2**i:
later = later - ((later - 1) % 2**i) - 1 later = later - ((later - 1) % 2**i) - 1
o = concat_generalized_indices(o, get_generalized_index(ShardState, 'history_acc', i)) o = concat_generalized_indices(o, get_generalized_index(ShardState, ['history_acc', i]))
return o return o
``` ```
@ -150,7 +150,7 @@ def get_generalized_index_of_crosslink_header(index: int) -> GeneralizedIndex:
#### `process_shard_receipt_proof` #### `process_shard_receipt_proof`
```python ```python
def process_shard_receipt_proof(state: BeaconState, receipt_proof: ShardReceiptProof): def process_shard_receipt_proof(state: BeaconState, receipt_proof: ShardReceiptProof) -> None:
""" """
Processes a ShardReceipt object. Processes a ShardReceipt object.
""" """
@ -221,8 +221,11 @@ def update_persistent_committee(state: BeaconState) -> None:
if (get_current_epoch(state) + 1) % EPOCHS_PER_SHARD_PERIOD == 0: if (get_current_epoch(state) + 1) % EPOCHS_PER_SHARD_PERIOD == 0:
period = (get_current_epoch(state) + 1) // EPOCHS_PER_SHARD_PERIOD period = (get_current_epoch(state) + 1) // EPOCHS_PER_SHARD_PERIOD
committees = Vector[CompactCommittee, SHARD_COUNT]([ committees = Vector[CompactCommittee, SHARD_COUNT]([
committee_to_compact_committee(state, get_period_committee(state, get_current_epoch(state) + 1, i)) committee_to_compact_committee(
for i in range(SHARD_COUNT) state,
get_period_committee(state, Epoch(get_current_epoch(state) + 1), Shard(shard)),
)
for shard in range(SHARD_COUNT)
]) ])
state.persistent_committee_roots[period % PERSISTENT_COMMITTEE_ROOT_LENGTH] = hash_tree_root(committees) state.persistent_committee_roots[period % PERSISTENT_COMMITTEE_ROOT_LENGTH] = hash_tree_root(committees)
``` ```

View File

@ -181,7 +181,7 @@ _Usage note: functions outside this section should manipulate generalized indice
#### `concat_generalized_indices` #### `concat_generalized_indices`
```python ```python
def concat_generalized_indices(indices: Sequence[GeneralizedIndex]) -> GeneralizedIndex: def concat_generalized_indices(*indices: GeneralizedIndex) -> GeneralizedIndex:
""" """
Given generalized indices i1 for A -> B, i2 for B -> C .... i_n for Y -> Z, returns Given generalized indices i1 for A -> B, i2 for B -> C .... i_n for Y -> Z, returns
the generalized index for A -> Z. the generalized index for A -> Z.