remove some generated_from_seed_indexed()

EspressoSystems · Oct 12, 2023 · 134d39a · 134d39a
1 parent 3fc6194
commit 134d39a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/crates/hotshot/examples/infra/modDA.rs b/crates/hotshot/examples/infra/modDA.rs
@@ -680,8 +680,7 @@ where
         let mut all_keys = BTreeSet::new();
         let mut da_keys = BTreeSet::new();
         for i in 0..config.config.total_nodes.get() as u64 {
-            let privkey = TYPES::SignatureKey::generated_from_seed_indexed([0u8; 32], i).1;
-            let pubkey = TYPES::SignatureKey::from_private(&privkey);
+            let pubkey = TYPES::SignatureKey::get_public_key(config.config.known_nodes_with_stake.get(&i).unwrap());
             if i < config.config.da_committee_size as u64 {
                 da_keys.insert(pubkey.clone());
             }

diff --git a/crates/testing/src/task_helpers.rs b/crates/testing/src/task_helpers.rs
@@ -51,7 +51,7 @@ pub async fn build_system_handle(
     let known_nodes_with_stake = config.known_nodes_with_stake.clone();
     let private_key =
         <BLSPubKey as SignatureKey>::generated_from_seed_indexed([0u8; 32], node_id).1;
-    let public_key = <SequencingTestTypes as NodeType>::SignatureKey::from_private(&private_key);
+    let public_key = <BLSPubKey as SignatureKey>::get_public_key(config.known_nodes_with_stake.get(node_id as usize).unwrap());
     let quorum_election_config = config.election_config.clone().unwrap_or_else(|| {
         <QuorumEx<SequencingTestTypes, SequencingMemoryImpl> as ConsensusExchange<
             SequencingTestTypes,