[CsvIO] update input Pcollection type arguments due to change in impl…

…ementation details. (#31891)
apache · Jul 17, 2024 · a767d41 · a767d41
1 parent ff15999
commit a767d41
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sdks/java/io/csv/src/main/java/org/apache/beam/sdk/io/csv/CsvIOParseKV.java b/sdks/java/io/csv/src/main/java/org/apache/beam/sdk/io/csv/CsvIOParseKV.java
@@ -30,11 +30,12 @@
  */
 // TODO(https://github.com/apache/beam/issues/31873): implement class after all dependencies are
 // completed.
-class CsvIOParseKV<T> extends PTransform<PCollection<KV<String, CSVRecord>>, PCollection<T>> {
+class CsvIOParseKV<T>
+    extends PTransform<PCollection<KV<String, Iterable<String>>>, PCollection<T>> {
 
   // TODO(https://github.com/apache/beam/issues/31873): implement method.
   @Override
-  public PCollection<T> expand(PCollection<KV<String, CSVRecord>> input) {
-    return input.apply(ParDo.of(new DoFn<KV<String, CSVRecord>, T>() {}));
+  public PCollection<T> expand(PCollection<KV<String, Iterable<String>>> input) {
+    return input.apply(ParDo.of(new DoFn<KV<String, Iterable<String>>, T>() {}));
   }
 }