Update h2o-core/src/main/java/water/parser/ParseDataset.java

Co-authored-by: Marek Novotný <[email protected]>
h2oai · Oct 25, 2023 · c21a63d · c21a63d
1 parent e4f2386
commit c21a63d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-core/src/main/java/water/parser/ParseDataset.java b/h2o-core/src/main/java/water/parser/ParseDataset.java
@@ -417,7 +417,7 @@ public int compare(ParseWriter.ParseErr o1, ParseWriter.ParseErr o2) {
       final int[] skippedColumns = setup.getSkippedColumns();
       String[] newColumnTypes;
       if (skippedColumns != null) {  // need to remove column types of skipped columns
-        List<Integer> skippedColIndices = Arrays.stream(skippedColumns).boxed().collect(Collectors.toList());
+        Set<Integer> skippedColIndices = Arrays.stream(skippedColumns).boxed().collect(Collectors.toSet());
         newColumnTypes = IntStream.range(0, originalColumnTypes.length).filter(x -> !(skippedColIndices.contains(x))).mapToObj(x -> originalColumnTypes[x]).toArray(String[]::new);
       } else {
         newColumnTypes = originalColumnTypes;