Ensure index_codecs chain is parsed correctly.

zoj613 · Jul 12, 2024 · e34b5d3 · e34b5d3
1 parent 90e769c
commit e34b5d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/codecs/array_to_bytes.ml b/lib/codecs/array_to_bytes.ml
@@ -233,7 +233,9 @@ end = struct
       Result.error @@ `Sharding (t.chunk_shape, repr.shape, msg))
     >>= fun () ->
     parse_chain repr t.codecs >>= fun () ->
-    parse_chain repr t.index_codecs
+    (* must add one dimension to the representation of the index array. *)
+    parse_chain
+      {repr with shape = Array.append repr.shape [|2|]} t.index_codecs
 
   let compute_encoded_size input_size t =
     List.fold_left BytesToBytes.compute_encoded_size