final fix: avoid mapping over the entire zkprogram

o1-labs · Dec 17, 2024 · bf1e275 · bf1e275
1 parent c9c81f7
commit bf1e275
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/lib/proof-system/recursive.ts b/src/lib/proof-system/recursive.ts
@@ -62,7 +62,7 @@ function Recursive<
 
   let methodKeys: MethodKey[] = Object.keys(methods);
 
-  let regularRecursiveProvers = mapObject(zkprogram, (prover, key) => {
+  let regularRecursiveProvers = mapToObject(methodKeys, (key) => {
     return async function proveRecursively_(
       publicInput: PublicInput,
       ...args: TupleToInstances<PrivateInputs[MethodKey]>
@@ -77,6 +77,9 @@ function Recursive<
         let constArgs = zip(args, privateInputs[key]).map(([arg, type]) =>
           Provable.toConstant(type, arg)
         );
+
+        let prover = zkprogram[key];
+
         if (hasPublicInput) {
           let { proof } = await prover(constInput, ...constArgs);
           return proof;