Merge branch 'dev' into JustinDrake-patch-14

scripts/phase0/build_spec.py

@@ -9,7 +9,6 @@ def build_phase0_spec(sourcefile, outfile):
 import copy
 from typing import (
     Any,
-    Callable,
     Dict,
     List,
     NewType,
@@ -18,19 +17,16 @@ from typing import (
 from eth2spec.utils.minimal_ssz import *
 from eth2spec.utils.bls_stub import *
 
-
-    """)
+""")
     for i in (1, 2, 3, 4, 8, 32, 48, 96):
         code_lines.append("def int_to_bytes%d(x): return x.to_bytes(%d, 'little')" % (i, i))
 
     code_lines.append("""
+
 # stub, will get overwritten by real var
 SLOTS_PER_EPOCH = 64
 
 
-def slot_to_epoch(x): return x // SLOTS_PER_EPOCH
-
-
 Slot = NewType('Slot', int)  # uint64
 Epoch = NewType('Epoch', int)  # uint64
 Shard = NewType('Shard', int)  # uint64
@@ -39,31 +35,24 @@ Gwei = NewType('Gwei', int)  # uint64
 Bytes32 = NewType('Bytes32', bytes)  # bytes32
 BLSPubkey = NewType('BLSPubkey', bytes)  # bytes48
 BLSSignature = NewType('BLSSignature', bytes)  # bytes96
-Any = None
 Store = None
-    """)
+""")
 
     code_lines += function_puller.get_spec(sourcefile)
 
     code_lines.append("""
-# Monkey patch validator get committee code
+# Monkey patch validator compute committee code
 _compute_committee = compute_committee
 committee_cache = {}
 
 
-def compute_committee(validator_indices: List[ValidatorIndex],
-                      seed: Bytes32,
-                      index: int,
-                      total_committees: int) -> List[ValidatorIndex]:
-
-    param_hash = (hash_tree_root(validator_indices), seed, index, total_committees)
+def compute_committee(indices: List[ValidatorIndex], seed: Bytes32, index: int, count: int) -> List[ValidatorIndex]:
+    param_hash = (hash_tree_root(indices), seed, index, count)
 
     if param_hash in committee_cache:
-        # print("Cache hit, epoch={0}".format(epoch))
         return committee_cache[param_hash]
     else:
-        # print("Cache miss, epoch={0}".format(epoch))
-        ret = _compute_committee(validator_indices, seed, index, total_committees)
+        ret = _compute_committee(indices, seed, index, count)
         committee_cache[param_hash] = ret
         return ret
 

scripts/phase0/function_puller.py

@@ -62,9 +62,10 @@ def get_spec(file_name: str) -> List[str]:
         code_lines.append('')
         for type_line in ssz_type:
             code_lines.append('    ' + type_line)
-    code_lines.append('')
+    code_lines.append('\n')
     code_lines.append('ssz_types = [' + ', '.join([f'\'{ssz_type_name}\'' for (ssz_type_name, _) in type_defs]) + ']')
-    code_lines.append('')
-    code_lines.append('def get_ssz_type_by_name(name: str) -> SSZType: return globals()[name]')
+    code_lines.append('\n')
+    code_lines.append('def get_ssz_type_by_name(name: str) -> SSZType:')
+    code_lines.append('    return globals()[name]')
     code_lines.append('')
     return code_lines