Skip to content

Commit dfc05b4

Browse files
committed
fix: correct type signature and non-unboxable types in applyDictWithRepStitch
- Add missing `Int` parameter (maxDef) to the type signature - Use `map (ds V.!) (VU.toList idxs)` for DInt96 and DText, since UTCTime and Text have no VU.Unbox instance and cannot be produced by VU.map
1 parent ee39bb9 commit dfc05b4

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/DataFrame/IO/Parquet/Dictionary.hs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,7 @@ applyDictWithRepStitch ::
197197
DictVals ->
198198
VU.Vector Int ->
199199
Int ->
200+
Int ->
200201
[Int] ->
201202
[Int] ->
202203
DI.Column
@@ -209,13 +210,13 @@ applyDictWithRepStitch dictVals idxs maxRep maxDef repLvls defLvls =
209210
DInt64 ds ->
210211
stitchForRepInt64 maxRep maxDef repLvls defLvls (VU.toList (VU.map (ds V.!) idxs))
211212
DInt96 ds ->
212-
stitchForRepUTCTime maxRep maxDef repLvls defLvls (VU.toList (VU.map (ds V.!) idxs))
213+
stitchForRepUTCTime maxRep maxDef repLvls defLvls (map (ds V.!) (VU.toList idxs))
213214
DFloat ds ->
214215
stitchForRepFloat maxRep maxDef repLvls defLvls (VU.toList (VU.map (ds V.!) idxs))
215216
DDouble ds ->
216217
stitchForRepDouble maxRep maxDef repLvls defLvls (VU.toList (VU.map (ds V.!) idxs))
217218
DText ds ->
218-
stitchForRepText maxRep maxDef repLvls defLvls (VU.toList (VU.map (ds V.!) idxs))
219+
stitchForRepText maxRep maxDef repLvls defLvls (map (ds V.!) (VU.toList idxs))
219220

220221
decodeDictV1 ::
221222
Maybe DictVals ->

0 commit comments

Comments
 (0)