Fix proof indexes (#351)

* Fix proof indexes * Upd packages
stakewise · Jul 6, 2024 · 777fab1 · 777fab1
1 parent b40b430
commit 777fab1
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 50 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/validators/execution.py b/src/validators/execution.py
@@ -308,7 +308,7 @@ async def register_multiple_validator(
     register_via_vault_v2 = False
 
     if multi_proof:
-        proof_indexes = calc_proof_indexes(multi_proof, tx_validators)
+        proof_indexes = [leaf[1] for leaf in multi_proof.leaves]
 
         # DepositDataRegistry args
         register_call_args = [
@@ -344,8 +344,3 @@ async def register_multiple_validator(
         logger.error('Registration transaction failed')
         return None
     return tx_hash
-
-
-def calc_proof_indexes(multi_proof: MultiProof, tx_validators: list[bytes]) -> list[int]:
-    sorted_tx_validators: list[bytes] = [v[0] for v in multi_proof.leaves]
-    return [sorted_tx_validators.index(v) for v in tx_validators]
diff --git a/src/validators/tasks.py b/src/validators/tasks.py
@@ -24,7 +24,6 @@
 from src.validators.database import NetworkValidatorCrud
 from src.validators.execution import (
     NetworkValidatorsProcessor,
-    calc_proof_indexes,
     get_latest_network_validator_public_keys,
     get_validators_from_deposit_data,
     get_validators_from_relayer,
@@ -163,7 +162,7 @@ async def register_validators(
             tree=cast(DepositData, deposit_data).tree,
             validators=cast(list[DepositDataValidator], validators),
         )
-        proof_indexes = calc_proof_indexes(multi_proof, tx_validators)
+        proof_indexes = [leaf[1] for leaf in multi_proof.leaves]
 
     else:
         tx_validators = [