fixed lint

Author: daniel-sanche

google/cloud/firestore_v1/base_pipeline.py

@@ -357,7 +357,7 @@ def sample(self, limit_or_options: int | SampleOptions) -> "_BasePipeline":
         """
         return self._append(stages.Sample(limit_or_options))
 
-    def union(self, other: Self) -> "_BasePipeline":
+    def union(self, other: "_BasePipeline") -> "_BasePipeline":
         """
         Performs a union of all documents from this pipeline and another pipeline,
         including duplicates.
@@ -384,7 +384,7 @@ def unnest(
         self,
         field: str | Selectable,
         alias: str | Field | None = None,
-        options: Optional[stages.UnnestOptions] = None,
+        options: stages.UnnestOptions | None = None,
     ) -> "_BasePipeline":
         """
         Produces a document for each element in an array field from the previous stage document.

google/cloud/firestore_v1/pipeline_expressions.py

@@ -15,23 +15,16 @@
 from __future__ import annotations
 from typing import (
     Any,
-    Iterable,
     List,
-    Mapping,
-    Union,
     Generic,
     TypeVar,
-    List,
     Dict,
-    Tuple,
     Sequence,
 )
 from abc import ABC
 from abc import abstractmethod
 from enum import Enum
-from enum import auto
 import datetime
-from dataclasses import dataclass
 from google.cloud.firestore_v1.types.document import Value
 from google.cloud.firestore_v1.types.query import StructuredQuery as Query_pb
 from google.cloud.firestore_v1.vector import Vector

google/cloud/firestore_v1/pipeline_source.py

@@ -82,4 +82,4 @@ def documents(self, *docs: "BaseDocumentReference") -> PipelineType:
         Returns:
             a new pipeline instance targeting the specified documents
         """
-        return self._create_pipeline(stages.Documents.of(*docs))
+        return self._create_pipeline(stages.Documents.of(*docs))

google/cloud/firestore_v1/pipeline_stages.py

@@ -33,7 +33,7 @@
 )
 
 if TYPE_CHECKING:
-    from google.cloud.firestore_v1.pipeline import Pipeline
+    from google.cloud.firestore_v1.base_pipeline import _BasePipeline
     from google.cloud.firestore_v1.base_document import BaseDocumentReference
 
 
@@ -369,7 +369,7 @@ def _pb_args(self):
 class Union(Stage):
     """Performs a union of documents from two pipelines."""
 
-    def __init__(self, other: Pipeline):
+    def __init__(self, other: _BasePipeline):
         super().__init__()
         self.other = other
 

tests/system/test_pipeline_acceptance.py

@@ -13,20 +13,17 @@
 # limitations under the License.
 
 from __future__ import annotations
-import sys
 import os
 import pytest
 import yaml
 import re
 from typing import Any
-from contextlib import nullcontext
 
 from google.protobuf.json_format import MessageToDict
 
 # from google.cloud.firestore_v1.pipeline_stages import *
 from google.cloud.firestore_v1 import pipeline_stages
 from google.cloud.firestore_v1 import pipeline_expressions
-from google.cloud.firestore_v1.pipeline import Pipeline
 from google.api_core.exceptions import GoogleAPIError
 
 from google.cloud.firestore import Client, AsyncClient