add stop iteration to close

apache · Oct 9, 2023 · 94b86c2 · 94b86c2
1 parent 8fc8e51
commit 94b86c2
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
@@ -1317,7 +1317,7 @@ def __next__(self):
             self.read_rows_response.avro_rows.serialized_binary_rows)
         return fastavro.schemaless_reader(self.bytes_reader, self.avro_schema)
       else:
-        raise EOFError
+        raise StopIteration
 
 
 @deprecated(since='2.11.0', current="WriteToBigQuery")