fix: add missing read api retry setting on SplitReadStream (#311)

- [ ] Regenerate this pull request now.

Committer: @yirutang
PiperOrigin-RevId: 398057257

Source-Link: googleapis/googleapis@376e8e9

Source-Link: https://github.com/googleapis/googleapis-gen/commit/32c05ed4ed88d83ad5cafc7544058acc58bf5628
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMzJjMDVlZDRlZDg4ZDgzYWQ1Y2FmYzc1NDQwNThhY2M1OGJmNTYyOCJ9

google/cloud/bigquery_storage_v1/services/big_query_read/async_client.py

@@ -441,7 +441,17 @@ async def split_read_stream(
         # and friendly error handling.
         rpc = gapic_v1.method_async.wrap_method(
             self._client._transport.split_read_stream,
-            default_timeout=None,
+            default_retry=retries.Retry(
+                initial=0.1,
+                maximum=60.0,
+                multiplier=1.3,
+                predicate=retries.if_exception_type(
+                    core_exceptions.DeadlineExceeded,
+                    core_exceptions.ServiceUnavailable,
+                ),
+                deadline=600.0,
+            ),
+            default_timeout=600.0,
             client_info=DEFAULT_CLIENT_INFO,
         )
 

google/cloud/bigquery_storage_v1/services/big_query_read/transports/base.py

@@ -189,7 +189,19 @@ def _prep_wrapped_messages(self, client_info):
                 client_info=client_info,
             ),
             self.split_read_stream: gapic_v1.method.wrap_method(
-                self.split_read_stream, default_timeout=None, client_info=client_info,
+                self.split_read_stream,
+                default_retry=retries.Retry(
+                    initial=0.1,
+                    maximum=60.0,
+                    multiplier=1.3,
+                    predicate=retries.if_exception_type(
+                        core_exceptions.DeadlineExceeded,
+                        core_exceptions.ServiceUnavailable,
+                    ),
+                    deadline=600.0,
+                ),
+                default_timeout=600.0,
+                client_info=client_info,
             ),
         }