Remove warning in dataframes io. (#29084)

apache · Oct 20, 2023 · 7610d46 · 7610d46
1 parent 641b95d
commit 7610d46
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/dataframe/io.py b/sdks/python/apache_beam/dataframe/io.py
@@ -281,8 +281,9 @@ def expand(self, root):
       if not self.binary:
         handle = TextIOWrapper(handle)
       if self.incremental:
-        sample = next(
-            self.reader(handle, *self.args, **dict(self.kwargs, chunksize=100)))
+        with self.reader(handle, *self.args, **dict(self.kwargs,
+                                                    chunksize=100)) as stream:
+          sample = next(stream)
       else:
         sample = self.reader(handle, *self.args, **self.kwargs)