Catch EOFError along with StopIteration (apache#28896)

* catch EOF instead of StopIteration * add stop iteration to close * catch both stop iteration and EOF
riteshghorse · Oct 9, 2023 · e815ad2 · e815ad2
1 parent 33b4168
commit e815ad2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/io/gcp/bigquery.py b/sdks/python/apache_beam/io/gcp/bigquery.py
@@ -1310,7 +1310,7 @@ def __iter__(self):
   def __next__(self):
     try:
       return fastavro.schemaless_reader(self.bytes_reader, self.avro_schema)
-    except StopIteration:
+    except (StopIteration, EOFError):
       self.read_rows_response = next(self.read_rows_iterator, None)
       if self.read_rows_response is not None:
         self.bytes_reader = io.BytesIO(