catch EOF instead of StopIteration

apache · Oct 9, 2023 · 8fc8e51 · 8fc8e51
1 parent 92d5bf8
commit 8fc8e51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/io/gcp/bigquery.py b/sdks/python/apache_beam/io/gcp/bigquery.py
@@ -1310,14 +1310,14 @@ def __iter__(self):
   def __next__(self):
     try:
       return fastavro.schemaless_reader(self.bytes_reader, self.avro_schema)
-    except StopIteration:
+    except EOFError:
       self.read_rows_response = next(self.read_rows_iterator, None)
       if self.read_rows_response is not None:
         self.bytes_reader = io.BytesIO(
             self.read_rows_response.avro_rows.serialized_binary_rows)
         return fastavro.schemaless_reader(self.bytes_reader, self.avro_schema)
       else:
-        raise StopIteration
+        raise EOFError
 
 
 @deprecated(since='2.11.0', current="WriteToBigQuery")