Flush the underlying file-like object.

apache · Sep 17, 2024 · ec37750 · ec37750
1 parent 8088ecf
commit ec37750
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sdks/python/apache_beam/dataframe/io.py b/sdks/python/apache_beam/dataframe/io.py
@@ -578,7 +578,7 @@ def _read(self, size=-1):
     return res
 
   def flush(self):
-    return
+    self._underlying.flush()
 
 
 class _ReadFromPandasDoFn(beam.DoFn, beam.RestrictionProvider):

diff --git a/sdks/python/apache_beam/io/textio_test.py b/sdks/python/apache_beam/io/textio_test.py
@@ -1737,8 +1737,8 @@ def test_non_utf8_csv_read_write(self):
       with TestPipeline() as p:
         r1 = (
             p
-            | beam.io.ReadFromCsv(input_fn, encoding="latin1")
-            | beam.Map(lambda x: x._asdict()))
+            | 'Read' >> beam.io.ReadFromCsv(input_fn, encoding="latin1")
+            | 'ToDict' >> beam.Map(lambda x: x._asdict()))
         assert_that(
             r1,
             equal_to([{
@@ -1750,7 +1750,7 @@ def test_non_utf8_csv_read_write(self):
       with TestPipeline() as p:
         _ = (
             p
-            | beam.io.ReadFromCsv(input_fn, encoding="latin1")
+            | 'Read' >> beam.io.ReadFromCsv(input_fn, encoding="latin1")
             | 'Write' >> beam.io.WriteToCsv(
                 os.path.join(dest, 'out'), encoding="latin1"))