diff --git a/packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py b/packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py index 02626dbef44a..1c902db10d31 100644 --- a/packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py +++ b/packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py @@ -1596,9 +1596,10 @@ async def test_call_metadata(self, include_app_profile, fn_name, fn_args, gapic_ # incomplete mutation, so an empty mock stream would retry until the # operation timeout. Return a success entry for the single mutation # so the operation completes after a single attempt. - from google.cloud.bigtable_v2.types import MutateRowsResponse from google.rpc import status_pb2 + from google.cloud.bigtable_v2.types import MutateRowsResponse + @CrossSync.convert async def mutate_rows_stream(*args, **kwargs): yield MutateRowsResponse(