fix: resolve issue AttributeError in bigquery_storage writer

googleapis · Mar 14, 2022 · 3956003 · 3956003
1 parent 07ba231
commit 3956003
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/google/cloud/bigquery_storage_v1/writer.py b/google/cloud/bigquery_storage_v1/writer.py
@@ -182,7 +182,7 @@ def _open(
  # ValueError: Can not send() on an RPC that has never been open()ed.
  #
  # when they try to send a request.
- while not self._rpc.is_active and self._consumer.is_active:
+ while self._rpc and not self._rpc.is_active and self._consumer and self._consumer.is_active:
  # Avoid 100% CPU while waiting for RPC to be ready.
  time.sleep(_WRITE_OPEN_INTERVAL)
 

diff --git a/google/cloud/bigquery_storage_v1beta2/writer.py b/google/cloud/bigquery_storage_v1beta2/writer.py
@@ -182,7 +182,7 @@ def _open(
  # ValueError: Can not send() on an RPC that has never been open()ed.
  #
  # when they try to send a request.
- while not self._rpc.is_active and self._consumer.is_active:
+ while self._rpc and not self._rpc.is_active and self._consumer and self._consumer.is_active:
  # Avoid 100% CPU while waiting for RPC to be ready.
  time.sleep(_WRITE_OPEN_INTERVAL)