fix DataFrame.flattenIndices test failure

rapidsai · Jun 2, 2023 · f4806cb · f4806cb
1 parent 218074f
commit f4806cb
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/modules/cudf/src/data_frame.ts b/modules/cudf/src/data_frame.ts
@@ -719,12 +719,11 @@ export class DataFrame<T extends TypeMap = any> {
                  const table = includeNulls ? df.asTable().explodeOuterPosition(i, mr)  //
                                             : df.asTable().explodePosition(i, mr);
                  return new DataFrame(df.names.reduce((series_map, name, index) => {
-                   if (index <= i) {
-                     series_map[name] =
-                       (this.get(name) as any).elements.__construct(table.getColumnByIndex(index));
+                   if (index === i) {
+                     series_map[name] = Series.new(table.getColumnByIndex(index));
                    } else {
                      series_map[name] =
-                       df.__constructChild(name, table.getColumnByIndex(index + 1));
+                       df.__constructChild(name, table.getColumnByIndex(+(index >= i) + index));
                    }
                    return series_map;
                  }, {} as SeriesMap<U>));