store witness JSON in a separate variable

privacy-scaling-explorations · Nov 4, 2023 · ba932e5 · ba932e5
1 parent f1e7491
commit ba932e5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/frontend/python/chiquito/dsl.py b/src/frontend/python/chiquito/dsl.py
@@ -73,15 +73,15 @@ def gen_witness(self: SuperCircuit, *args: Any) -> Dict[int, TraceWitness]:
     def halo2_mock_prover(
         self: SuperCircuit, super_witness: Dict[int, TraceWitness], k: int = 16
     ):
+        witness_json = {}
         for rust_id, witness in super_witness.items():
             if rust_id not in self.ast.sub_circuits:
                 raise ValueError(
                     f"SuperCircuit.halo2_mock_prover(): TraceWitness with rust_id {rust_id} not found in sub_circuits."
                 )
-            witness_json: str = witness.get_witness_json()
-            super_witness[rust_id] = witness_json
+            witness_json[rust_id] = witness.get_witness_json()
         rust_chiquito.super_circuit_halo2_mock_prover(
-            list(self.ast.sub_circuits.keys()), super_witness, k
+            list(self.ast.sub_circuits.keys()), witness_json, k
         )