From 724352e19f91254f304b98457e066389d39c06a5 Mon Sep 17 00:00:00 2001
From: danielSanchezQ <3danimanimal@gmail.com>
Date: Fri, 24 Jan 2025 16:51:12 +0000
Subject: [PATCH] Fix missing parameter in dispersal

---
 da/dispersal.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/da/dispersal.py b/da/dispersal.py
index c5e6016..f9f9f5c 100644
--- a/da/dispersal.py
+++ b/da/dispersal.py
@@ -21,14 +21,19 @@ class Dispersal:
     def _prepare_data(self, encoded_data: EncodedData) -> Generator[DAShare, None, None]:
         columns = encoded_data.extended_matrix.columns
         row_commitments = encoded_data.row_commitments
-        column_proofs = encoded_data.combined_column_proofs
-        blobs_data = zip(columns, column_proofs)
-        for column_idx, (column, proof) in enumerate(blobs_data):
-            blob = DAShare(
-                Column(column),
+        rows_proofs = encoded_data.row_proofs
+        aggregated_column_commitment = encoded_data.aggregated_column_commitment
+        aggregated_column_proofs = encoded_data.aggregated_column_proofs
+        blobs_data = zip(columns, column_commitments, zip(*rows_proofs), aggregated_column_proofs)
+        for column_idx, (column, column_commitment, row_proofs, column_proof) in enumerate(blobs_data):
+            blob = DABlob(
+                column,
                 column_idx,
-                proof,
-                row_commitments
+                column_commitment,
+                aggregated_column_commitment,
+                column_proof,
+                row_commitments,
+                row_proofs
             )
             yield blob