Update bigquery/tests/integration/smoke_test.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: TheKevJames

bigquery/tests/integration/smoke_test.py

@@ -67,16 +67,12 @@ def is_processing(operation) -> bool:
                 and operation.metadata['common']['state'] == 'PROCESSING'
             )
 
-        @retry(
+        operation = await retry(
             retry=retry_if_result(is_processing),
             wait=wait_fixed(10),
             stop=stop_after_attempt(10),
             reraise=True,
-        )
-        async def get_operation_with_retry(name: str):
-            return await ds.get_datastore_operation(name)
-
-        operation = await get_operation_with_retry(operation.name)
+        )(ds.get_datastore_operation)(operation.name)
 
         assert operation.metadata['common']['state'] == 'SUCCESSFUL'
         # END: copy from `test_datastore_export`