Dont re-use pandas sink instances across windows

apache · Mar 30, 2021 · b88ebc8 · b88ebc8
1 parent 949d2e0
commit b88ebc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/dataframe/io.py b/sdks/python/apache_beam/dataframe/io.py
@@ -521,7 +521,7 @@ def expand(self, pcoll):
     return pcoll | fileio.WriteToFiles(
         path=dir,
         file_naming=fileio.default_file_naming(name),
-        sink=_WriteToPandasFileSink(
+        sink=lambda _: _WriteToPandasFileSink(
             self.writer, self.args, self.kwargs, self.incremental, self.binary))