From a8d79c8521d767e159952010f1ded66c3e5cf9bc Mon Sep 17 00:00:00 2001 From: Jeffrey Yasskin Date: Wed, 9 Nov 2022 19:54:06 -0800 Subject: [PATCH] fix: Fix typehint on AsyncDocumentReference.get() (#649) * fix: AsyncDocumentReference.get() only returns 1 level of coroutine. * Remove no-longer-used imports. Co-authored-by: Mariatta Wijaya --- google/cloud/firestore_v1/async_document.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/google/cloud/firestore_v1/async_document.py b/google/cloud/firestore_v1/async_document.py index a6606963e3..47cce42af8 100644 --- a/google/cloud/firestore_v1/async_document.py +++ b/google/cloud/firestore_v1/async_document.py @@ -28,7 +28,7 @@ from google.cloud.firestore_v1 import _helpers from google.cloud.firestore_v1.types import write from google.protobuf.timestamp_pb2 import Timestamp -from typing import Any, AsyncGenerator, Coroutine, Iterable, Union +from typing import AsyncGenerator, Iterable logger = logging.getLogger(__name__) @@ -329,7 +329,7 @@ async def get( transaction=None, retry: retries.Retry = gapic_v1.method.DEFAULT, timeout: float = None, - ) -> Union[DocumentSnapshot, Coroutine[Any, Any, DocumentSnapshot]]: + ) -> DocumentSnapshot: """Retrieve a snapshot of the current document. See :meth:`~google.cloud.firestore_v1.base_client.BaseClient.field_path` for