Fix: ProofInputSample data is cell bytes instead of block bytes

codex/slots/builder/builder.nim

@@ -127,6 +127,13 @@ func numBlockCells*(self: SlotsBuilder): Natural =
 
   (self.manifest.blockSize div self.cellSize).Natural
 
+func cellSize*(self: SlotsBuilder): NBytes =
+  ## Cell size.
+  ##
+
+  self.cellSize
+
+
 func numSlotCells*(self: SlotsBuilder): Natural =
   ## Number of cells per slot.
   ##

codex/slots/sampler/sampler.nim

@@ -69,7 +69,14 @@ proc new*(
     blockStore: blockStore,
     builder: builder)
 
-proc getProofs*(
+proc getCell*(self: DataSampler, blkBytes: seq[byte], blkCellIdx: uint64): Cell =
+  let
+    cellSize = self.builder.cellSize.uint64
+    dataStart = (cellSize * blkCellIdx)
+    dataEnd = dataStart + cellSize
+  return blkBytes[dataStart ..< dataEnd]
+
+proc getProofInput*(
   self: DataSampler,
   entropy: ProofChallenge,
   nSamples: Natural): Future[?!ProofInput] {.async.} =
@@ -126,7 +133,9 @@ proc getProofs*(
         error "Failed to get proof from block tree", err = err.msg
         return failure(err)
 
-      Sample(data: bytes, proof: blockProof)
+      let cellData = self.getCell(bytes, blkCellIdx)
+
+      Sample(data: cellData, proof: blockProof)
 
   success ProofInput(
     entropy: entropy,