From dea70b724ae7fa637484dfa97f104cf60ead8220 Mon Sep 17 00:00:00 2001
From: mmagician <marcin.gorny.94@protonmail.com>
Date: Fri, 26 Jan 2024 14:20:44 +0100
Subject: [PATCH] prover state should include full randomness

for "better used as a part of a larger protocol." as the docs say
---
 src/ml_sumcheck/mod.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/ml_sumcheck/mod.rs b/src/ml_sumcheck/mod.rs
index 4fd2f9f..735f0ab 100644
--- a/src/ml_sumcheck/mod.rs
+++ b/src/ml_sumcheck/mod.rs
@@ -62,6 +62,9 @@ impl<F: Field> MLSumcheck<F> {
             prover_msgs.push(prover_msg);
             verifier_msg = Some(IPForMLSumcheck::sample_round(fs_rng));
         }
+        prover_state
+            .randomness
+            .push(verifier_msg.unwrap().randomness);
 
         Ok((prover_msgs, prover_state))
     }