update_pending_votes: search pending header by header.(root + slot + shard)

This commit is contained in:
Anton Nashatyrev 2021-05-05 13:31:19 +03:00
parent befe4c7db3
commit 3cc1256a72
1 changed files with 12 additions and 7 deletions

View File

@ -512,17 +512,22 @@ def update_pending_votes(state: BeaconState, attestation: Attestation) -> None:
pending_headers = state.current_epoch_pending_shard_headers pending_headers = state.current_epoch_pending_shard_headers
else: else:
pending_headers = state.previous_epoch_pending_shard_headers pending_headers = state.previous_epoch_pending_shard_headers
pending_header = None
for header in pending_headers: attestation_shard = compute_shard_from_committee_index(
if header.root == attestation.data.shard_header_root:
pending_header = header
assert pending_header is not None
assert pending_header.slot == attestation.data.slot
assert pending_header.shard == compute_shard_from_committee_index(
state, state,
attestation.data.slot, attestation.data.slot,
attestation.data.index, attestation.data.index,
) )
pending_header = None
for header in pending_headers:
if (
header.root == attestation.data.shard_header_root
and header.slot == attestation.data.slot
and header.shard == attestation_shard
):
pending_header = header
assert pending_header is not None
for i in range(len(pending_header.votes)): for i in range(len(pending_header.votes)):
pending_header.votes[i] = pending_header.votes[i] or attestation.aggregation_bits[i] pending_header.votes[i] = pending_header.votes[i] or attestation.aggregation_bits[i]