transaction return type hint

googleapis · Aug 7, 2024 · 6f86854 · 6f86854
1 parent bb60ce0
commit 6f86854
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: