fixed mypy

Author: daniel-sanche

google/cloud/firestore_v1/base_pipeline.py

@@ -105,6 +105,6 @@ def _execute_response_helper(
                 doc.fields,
                 ref,
                 response._pb.execution_time,
-                doc.create_time.timestamp_pb() if doc.create_time else None,
-                doc.update_time.timestamp_pb() if doc.update_time else None,
+                doc._pb.create_time if doc.create_time else None,
+                doc._pb.update_time if doc.update_time else None,
             )

google/cloud/firestore_v1/field_path.py

@@ -16,7 +16,7 @@
 from __future__ import annotations
 import re
 from collections import abc
-from typing import Iterable, cast
+from typing import Any, Iterable, cast, MutableMapping
 
 _FIELD_PATH_MISSING_TOP = "{!r} is not contained in the data"
 _FIELD_PATH_MISSING_KEY = "{!r} is not contained in the data for the key {!r}"
@@ -170,7 +170,7 @@ def render_field_path(field_names: Iterable[str]):
 get_field_path = render_field_path  # backward-compatibility
 
 
-def get_nested_value(field_path: str, data: dict):
+def get_nested_value(field_path: str, data: MutableMapping[str, Any]):
     """Get a (potentially nested) value from a dictionary.
 
     If the data is nested, for example:

google/cloud/firestore_v1/pipeline_result.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 from __future__ import annotations
-from typing import Any, TYPE_CHECKING
+from typing import Any, MutableMapping, TYPE_CHECKING
 from google.cloud.firestore_v1 import _helpers
 from google.cloud.firestore_v1.field_path import get_nested_value
 from google.cloud.firestore_v1.field_path import FieldPath
@@ -36,7 +36,7 @@ class PipelineResult:
     def __init__(
         self,
         client: BaseClient,
-        fields_pb: dict[str, ValueProto],
+        fields_pb: MutableMapping[str, ValueProto],
         ref: BaseDocumentReference | None = None,
         execution_time: Timestamp | None = None,
         create_time: Timestamp | None = None,