From 6f8685412bf312eb6bbef7f56e08d1bf15ddf70f Mon Sep 17 00:00:00 2001 From: Linchin Date: Tue, 6 Aug 2024 17:06:39 -0700 Subject: [PATCH] transaction return type hint --- google/cloud/firestore_v1/transaction.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/google/cloud/firestore_v1/transaction.py b/google/cloud/firestore_v1/transaction.py index ff1022c81..ea22ab9a1 100644 --- a/google/cloud/firestore_v1/transaction.py +++ b/google/cloud/firestore_v1/transaction.py @@ -44,6 +44,7 @@ if TYPE_CHECKING: # pragma: NO COVER from google.cloud.firestore_v1.base_document import DocumentSnapshot from google.cloud.firestore_v1.query_profile import ExplainOptions + from google.cloud.firestore_v1.stream_generator import StreamGenerator from google.cloud.firestore_v1.types import CommitResponse @@ -176,7 +177,7 @@ def get( timeout: float = None, *, explain_options: Optional[ExplainOptions] = None, - ) -> Generator[DocumentSnapshot, Any, None]: + ) -> StreamGenerator[DocumentSnapshot]: """Retrieve a document or a query result from the database. Args: