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: