remove unneeded except

Author: shuoweil

bigframes/display/anywidget.py

@@ -21,7 +21,6 @@
 from typing import Any, cast, Dict, Iterator, List, Optional, Type
 import uuid
 
-import google.api_core.exceptions
 import pandas as pd
 
 import bigframes
@@ -168,12 +167,7 @@ def _get_next_batch(self) -> bool:
             batch = next(iterator)
             self._cached_batches.append(batch)
             return True
-        except (
-            StopIteration,
-            google.api_core.exceptions.GoogleAPICallError,
-            TypeError,
-            ValueError,
-        ) as e:
+        except StopIteration as e:
             self._all_data_loaded = True
             if not isinstance(e, StopIteration):
                 # If we fail to get a batch, assume no more data is available.

tests/benchmark/read_gbq_colab/filter_output.py

@@ -37,6 +37,7 @@ def filter_output(
     # Simulate the user filtering by a column and visualizing those results
     df_filtered = df[df["col_bool_0"]]
     batches_filtered = df_filtered.to_pandas_batches(page_size=PAGE_SIZE)
+    assert batches_filtered.total_rows >= 0
 
     # It's possible we don't have any pages at all, since we filtered out all
     # matching rows.