Fix index var name

stakewise · Jan 30, 2023 · f273a30 · f273a30
1 parent 3b28632
commit f273a30
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/validators/execution.py b/src/validators/execution.py
@@ -228,7 +228,7 @@ async def register_single_validator(
 
     credentials = get_eth1_withdrawal_credentials(VAULT_CONTRACT_ADDRESS)
     tx_validator = _encode_tx_validator(credentials, validator)
-    proof = tree.get_proof([tx_validator, validator.index])  # type: ignore
+    proof = tree.get_proof([tx_validator, validator.deposit_data_index])  # type: ignore
 
     tx_data = SingleValidatorRegistration(
         keeperParams=KeeperApprovalParams(
@@ -268,7 +268,7 @@ async def register_multiple_validator(
     for validator in validators:
         tx_validator = _encode_tx_validator(credentials, validator)
         tx_validators.append(tx_validator)
-        leaves.append((tx_validator, validator.index))
+        leaves.append((tx_validator, validator.deposit_data_index))
 
     multi_proof = tree.get_multi_proof(leaves)
     sorted_tx_validators: list[bytes] = [v[0] for v in multi_proof.leaves]

diff --git a/src/validators/typings.py b/src/validators/typings.py
@@ -18,7 +18,7 @@ class NetworkValidator:
 
 @dataclass
 class Validator:
-    index: int
+    deposit_data_index: int
     public_key: HexStr
     signature: HexStr
 

diff --git a/src/validators/utils.py b/src/validators/utils.py
@@ -130,7 +130,7 @@ async def load_deposit_data() -> DepositData:
     validators: list[Validator] = []
     for i, data in enumerate(deposit_data):
         validator = Validator(
-            index=i,
+            deposit_data_index=i,
             public_key=add_0x_prefix(data['pubkey']),
             signature=add_0x_prefix(data['signature']),
         )