fix: ensure we pass the correct data to add tally results

privacy-scaling-explorations · Nov 5, 2024 · 77028ac · 77028ac
1 parent 6007edf
commit 77028ac
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/contracts/tasks/helpers/Prover.ts b/packages/contracts/tasks/helpers/Prover.ts
@@ -322,18 +322,20 @@ export class Prover {
     console.log("Submitting results...");
 
     const tallyResults = tallyData.results.tally.map((t) => BigInt(t));
+    const resultLength = recipients ?? tallyResults.length;
+
+    // slice in case we are submitting partial results
+    const partialResults = tallyResults.slice(0, resultLength);
 
     const [treeDepths] = await Promise.all([this.pollContract.treeDepths()]);
 
-    const resultLength = recipients ?? tallyResults.length;
-    const tallyResultProofs = tallyResults
-      .slice(0, resultLength)
+    const tallyResultProofs = partialResults
       .map((_, index) => genTreeProof(index, tallyResults, Number(treeDepths.voteOptionTreeDepth)));
 
     await this.tallyContract
       .addTallyResults({
-        voteOptionIndices: tallyData.results.tally.map((_, index) => index),
-        tallyResults,
+        voteOptionIndices: partialResults.map((_, index) => index),
+        tallyResults: partialResults,
         tallyResultProofs,
         totalSpent: tallyData.totalSpentVoiceCredits.spent,
         totalSpentSalt: tallyData.totalSpentVoiceCredits.salt,