apache-airflow-providers-google 10.13.0rc1__tar.gz → 10.13.0rc2__tar.gz

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-google
-Version: 10.13.0rc1
+Version: 10.13.0rc2
 Summary: Provider package apache-airflow-providers-google for Apache Airflow
 Keywords: airflow-provider,google,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -34,26 +34,26 @@ Requires-Dist: google-api-python-client>=1.6.0
 Requires-Dist: google-auth-httplib2>=0.0.1
 Requires-Dist: google-auth>=1.0.0
 Requires-Dist: google-cloud-aiplatform>=1.22.1
-Requires-Dist: google-cloud-automl>=2.11.0
+Requires-Dist: google-cloud-automl>=2.12.0
 Requires-Dist: google-cloud-batch>=0.13.0
-Requires-Dist: google-cloud-bigquery-datatransfer>=3.11.0
+Requires-Dist: google-cloud-bigquery-datatransfer>=3.13.0
 Requires-Dist: google-cloud-bigtable>=2.17.0
-Requires-Dist: google-cloud-build>=3.13.0
+Requires-Dist: google-cloud-build>=3.22.0
 Requires-Dist: google-cloud-compute>=1.10.0
 Requires-Dist: google-cloud-container>=2.17.4
 Requires-Dist: google-cloud-datacatalog>=3.11.1
-Requires-Dist: google-cloud-dataflow-client>=0.8.2
+Requires-Dist: google-cloud-dataflow-client>=0.8.6
 Requires-Dist: google-cloud-dataform>=0.5.0
-Requires-Dist: google-cloud-dataplex>=1.4.2
+Requires-Dist: google-cloud-dataplex>=1.10.0
 Requires-Dist: google-cloud-dataproc-metastore>=1.12.0
-Requires-Dist: google-cloud-dataproc>=5.5.0
+Requires-Dist: google-cloud-dataproc>=5.8.0
 Requires-Dist: google-cloud-dlp>=3.12.0
 Requires-Dist: google-cloud-kms>=2.15.0
 Requires-Dist: google-cloud-language>=2.9.0
 Requires-Dist: google-cloud-logging>=3.5.0
 Requires-Dist: google-cloud-memcache>=1.7.0
 Requires-Dist: google-cloud-monitoring>=2.14.1
-Requires-Dist: google-cloud-orchestration-airflow>=1.7.0
+Requires-Dist: google-cloud-orchestration-airflow>=1.10.0
 Requires-Dist: google-cloud-os-login>=2.9.1
 Requires-Dist: google-cloud-pubsub>=2.15.0
 Requires-Dist: google-cloud-redis>=2.12.0
@@ -167,7 +167,7 @@ Provides-Extra: trino
 
 Package ``apache-airflow-providers-google``
 
-Release: ``10.13.0.rc1``
+Release: ``10.13.0.rc2``
 
 
 Google services including:
@@ -217,17 +217,17 @@ PIP package                              Version required
 ``google-auth``                          ``>=1.0.0``
 ``google-auth-httplib2``                 ``>=0.0.1``
 ``google-cloud-aiplatform``              ``>=1.22.1``
-``google-cloud-automl``                  ``>=2.11.0``
-``google-cloud-bigquery-datatransfer``   ``>=3.11.0``
+``google-cloud-automl``                  ``>=2.12.0``
+``google-cloud-bigquery-datatransfer``   ``>=3.13.0``
 ``google-cloud-bigtable``                ``>=2.17.0``
-``google-cloud-build``                   ``>=3.13.0``
+``google-cloud-build``                   ``>=3.22.0``
 ``google-cloud-compute``                 ``>=1.10.0``
 ``google-cloud-container``               ``>=2.17.4``
 ``google-cloud-datacatalog``             ``>=3.11.1``
-``google-cloud-dataflow-client``         ``>=0.8.2``
+``google-cloud-dataflow-client``         ``>=0.8.6``
 ``google-cloud-dataform``                ``>=0.5.0``
-``google-cloud-dataplex``                ``>=1.4.2``
-``google-cloud-dataproc``                ``>=5.5.0``
+``google-cloud-dataplex``                ``>=1.10.0``
+``google-cloud-dataproc``                ``>=5.8.0``
 ``google-cloud-dataproc-metastore``      ``>=1.12.0``
 ``google-cloud-dlp``                     ``>=3.12.0``
 ``google-cloud-kms``                     ``>=2.15.0``
@@ -235,7 +235,7 @@ PIP package                              Version required
 ``google-cloud-logging``                 ``>=3.5.0``
 ``google-cloud-memcache``                ``>=1.7.0``
 ``google-cloud-monitoring``              ``>=2.14.1``
-``google-cloud-orchestration-airflow``   ``>=1.7.0``
+``google-cloud-orchestration-airflow``   ``>=1.10.0``
 ``google-cloud-os-login``                ``>=2.9.1``
 ``google-cloud-pubsub``                  ``>=2.15.0``
 ``google-cloud-redis``                   ``>=2.12.0``

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/README.rst

@@ -42,7 +42,7 @@
 
 Package ``apache-airflow-providers-google``
 
-Release: ``10.13.0.rc1``
+Release: ``10.13.0.rc2``
 
 
 Google services including:
@@ -92,17 +92,17 @@ PIP package                              Version required
 ``google-auth``                          ``>=1.0.0``
 ``google-auth-httplib2``                 ``>=0.0.1``
 ``google-cloud-aiplatform``              ``>=1.22.1``
-``google-cloud-automl``                  ``>=2.11.0``
-``google-cloud-bigquery-datatransfer``   ``>=3.11.0``
+``google-cloud-automl``                  ``>=2.12.0``
+``google-cloud-bigquery-datatransfer``   ``>=3.13.0``
 ``google-cloud-bigtable``                ``>=2.17.0``
-``google-cloud-build``                   ``>=3.13.0``
+``google-cloud-build``                   ``>=3.22.0``
 ``google-cloud-compute``                 ``>=1.10.0``
 ``google-cloud-container``               ``>=2.17.4``
 ``google-cloud-datacatalog``             ``>=3.11.1``
-``google-cloud-dataflow-client``         ``>=0.8.2``
+``google-cloud-dataflow-client``         ``>=0.8.6``
 ``google-cloud-dataform``                ``>=0.5.0``
-``google-cloud-dataplex``                ``>=1.4.2``
-``google-cloud-dataproc``                ``>=5.5.0``
+``google-cloud-dataplex``                ``>=1.10.0``
+``google-cloud-dataproc``                ``>=5.8.0``
 ``google-cloud-dataproc-metastore``      ``>=1.12.0``
 ``google-cloud-dlp``                     ``>=3.12.0``
 ``google-cloud-kms``                     ``>=2.15.0``
@@ -110,7 +110,7 @@ PIP package                              Version required
 ``google-cloud-logging``                 ``>=3.5.0``
 ``google-cloud-memcache``                ``>=1.7.0``
 ``google-cloud-monitoring``              ``>=2.14.1``
-``google-cloud-orchestration-airflow``   ``>=1.7.0``
+``google-cloud-orchestration-airflow``   ``>=1.10.0``
 ``google-cloud-os-login``                ``>=2.9.1``
 ``google-cloud-pubsub``                  ``>=2.15.0``
 ``google-cloud-redis``                   ``>=2.12.0``

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/fs/gcs.py

@@ -39,7 +39,7 @@ GCS_VERSION_AWARE = "gcs.version-aware"
 schemes = ["gs", "gcs"]
 
 
-def get_fs(conn_id: str | None) -> AbstractFileSystem:
+def get_fs(conn_id: str | None, storage_options: dict[str, str] | None = None) -> AbstractFileSystem:
     # https://gcsfs.readthedocs.io/en/latest/api.html#gcsfs.core.GCSFileSystem
     from gcsfs import GCSFileSystem
 
@@ -49,15 +49,18 @@ def get_fs(conn_id: str | None) -> AbstractFileSystem:
     g = GoogleBaseHook(gcp_conn_id=conn_id)
     creds = g.get_credentials()
 
-    return GCSFileSystem(
-        project=g.project_id,
-        access=g.extras.get(GCS_ACCESS, "full_control"),
-        token=creds.token,
-        consistency=g.extras.get(GCS_CONSISTENCY, "none"),
-        cache_timeout=g.extras.get(GCS_CACHE_TIMEOUT),
-        requester_pays=g.extras.get(GCS_REQUESTER_PAYS, False),
-        session_kwargs=g.extras.get(GCS_SESSION_KWARGS, {}),
-        endpoint_url=g.extras.get(GCS_ENDPOINT),
-        default_location=g.extras.get(GCS_DEFAULT_LOCATION),
-        version_aware=g.extras.get(GCS_VERSION_AWARE, "false").lower() == "true",
-    )
+    options = {
+        "project": g.project_id,
+        "access": g.extras.get(GCS_ACCESS, "full_control"),
+        "token": creds.token,
+        "consistency": g.extras.get(GCS_CONSISTENCY, "none"),
+        "cache_timeout": g.extras.get(GCS_CACHE_TIMEOUT),
+        "requester_pays": g.extras.get(GCS_REQUESTER_PAYS, False),
+        "session_kwargs": g.extras.get(GCS_SESSION_KWARGS, {}),
+        "endpoint_url": g.extras.get(GCS_ENDPOINT),
+        "default_location": g.extras.get(GCS_DEFAULT_LOCATION),
+        "version_aware": g.extras.get(GCS_VERSION_AWARE, "false").lower() == "true",
+    }
+    options.update(storage_options or {})
+
+    return GCSFileSystem(**options)

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/bigquery_dts.py

@@ -38,6 +38,7 @@ from airflow.providers.google.common.hooks.base_google import (
 
 if TYPE_CHECKING:
     from google.api_core.retry import Retry
+    from google.api_core.retry_async import AsyncRetry
     from googleapiclient.discovery import Resource
 
 
@@ -321,7 +322,7 @@ class AsyncBiqQueryDataTransferServiceHook(GoogleBaseAsyncHook):
         run_id: str,
         project_id: str | None,
         location: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ):

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/cloud_build.py

@@ -33,6 +33,7 @@ from airflow.providers.google.common.hooks.base_google import PROVIDE_PROJECT_ID
 if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.retry import Retry
+    from google.api_core.retry_async import AsyncRetry
     from google.cloud.devtools.cloudbuild_v1.types import Build, BuildTrigger, RepoSource
 
 # Time to sleep between active checks of the operation results
@@ -645,7 +646,7 @@ class CloudBuildAsyncHook(GoogleBaseHook):
         self,
         id_: str,
         project_id: str = PROVIDE_PROJECT_ID,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
         location: str = "global",

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/cloud_composer.py

@@ -35,6 +35,7 @@ if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.operation_async import AsyncOperation
     from google.api_core.retry import Retry
+    from google.api_core.retry_async import AsyncRetry
     from google.cloud.orchestration.airflow.service_v1.services.environments.pagers import (
         ListEnvironmentsPager,
     )
@@ -332,7 +333,7 @@ class CloudComposerAsyncHook(GoogleBaseHook):
         project_id: str,
         region: str,
         environment: Environment | dict,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -361,7 +362,7 @@ class CloudComposerAsyncHook(GoogleBaseHook):
         project_id: str,
         region: str,
         environment_id: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -389,7 +390,7 @@ class CloudComposerAsyncHook(GoogleBaseHook):
         environment_id: str,
         environment: Environment | dict,
         update_mask: dict | FieldMask,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/compute_ssh.py

@@ -96,8 +96,8 @@ class ComputeEngineSSHHook(SSHHook):
     conn_type = "gcpssh"
     hook_name = "Google Cloud SSH"
 
-    @staticmethod
-    def get_ui_field_behaviour() -> dict[str, Any]:
+    @classmethod
+    def get_ui_field_behaviour(cls) -> dict[str, Any]:
         return {
             "hidden_fields": ["host", "schema", "login", "password", "port", "extra"],
             "relabeling": {},
@@ -295,7 +295,7 @@ class ComputeEngineSSHHook(SSHHook):
                 client = _GCloudAuthorizedSSHClient(self._compute_hook)
                 # Default is RejectPolicy
                 # No known host checking since we are not storing privatekey
-                client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+                client.set_missing_host_key_policy(paramiko.AutoAddPolicy())  # nosec B507
                 client.connect(
                     hostname=hostname,
                     username=user,

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/dataflow.py

@@ -55,7 +55,7 @@ T = TypeVar("T", bound=Callable)
 
 
 def process_line_and_extract_dataflow_job_id_callback(
-    on_new_job_id_callback: Callable[[str], None] | None
+    on_new_job_id_callback: Callable[[str], None] | None,
 ) -> Callable[[str], None]:
     """Build callback that triggers the specified function.
 

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/dataplex.py

@@ -40,6 +40,7 @@ from airflow.providers.google.common.hooks.base_google import GoogleBaseAsyncHoo
 if TYPE_CHECKING:
     from google.api_core.operation import Operation
     from google.api_core.retry import Retry
+    from google.api_core.retry_async import AsyncRetry
     from googleapiclient.discovery import Resource
 
 PATH_DATA_SCAN = "projects/{project_id}/locations/{region}/dataScans/{data_scan_id}"
@@ -896,7 +897,7 @@ class DataplexAsyncHook(GoogleBaseAsyncHook):
         region: str,
         data_scan_id: str | None = None,
         job_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Any:

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/hooks/dataproc.py

@@ -51,6 +51,7 @@ if TYPE_CHECKING:
     from google.api_core.operation_async import AsyncOperation
     from google.api_core.operations_v1.operations_client import OperationsClient
     from google.api_core.retry import Retry
+    from google.api_core.retry_async import AsyncRetry
     from google.protobuf.duration_pb2 import Duration
     from google.protobuf.field_mask_pb2 import FieldMask
 
@@ -256,7 +257,7 @@ class DataprocHook(GoogleBaseHook):
         self,
         operation: Operation,
         timeout: float | None = None,
-        result_retry: Retry | _MethodDefault = DEFAULT,
+        result_retry: AsyncRetry | _MethodDefault = DEFAULT,
     ) -> Any:
         """Wait for a long-lasting operation to complete."""
         try:
@@ -997,7 +998,7 @@ class DataprocHook(GoogleBaseHook):
         region: str,
         project_id: str,
         wait_check_interval: int = 10,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Batch:
@@ -1132,7 +1133,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         virtual_cluster_config: dict | None = None,
         labels: dict[str, str] | None = None,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1199,7 +1200,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         project_id: str,
         cluster_uuid: str | None = None,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1242,7 +1243,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         region: str,
         cluster_name: str,
         project_id: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> str:
@@ -1277,7 +1278,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         region: str,
         cluster_name: str,
         project_id: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Cluster:
@@ -1309,7 +1310,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         filter_: str,
         project_id: str,
         page_size: int | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ):
@@ -1349,7 +1350,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         region: str,
         graceful_decommission_timeout: dict | Duration | None = None,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1429,7 +1430,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         template: dict | WorkflowTemplate,
         project_id: str,
         region: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> WorkflowTemplate:
@@ -1465,7 +1466,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         version: int | None = None,
         request_id: str | None = None,
         parameters: dict[str, str] | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1511,7 +1512,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         project_id: str,
         region: str,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1554,7 +1555,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         job_id: str,
         project_id: str,
         region: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Job:
@@ -1588,7 +1589,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         project_id: str,
         region: str,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Job:
@@ -1624,7 +1625,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         job_id: str,
         project_id: str,
         region: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Job:
@@ -1658,7 +1659,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         batch: dict | Batch,
         batch_id: str | None = None,
         request_id: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AsyncOperation:
@@ -1703,7 +1704,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         batch_id: str,
         region: str,
         project_id: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
@@ -1737,7 +1738,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         batch_id: str,
         region: str,
         project_id: str,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Batch:
@@ -1773,7 +1774,7 @@ class DataprocAsyncHook(GoogleBaseHook):
         project_id: str,
         page_size: int | None = None,
         page_token: str | None = None,
-        retry: Retry | _MethodDefault = DEFAULT,
+        retry: AsyncRetry | _MethodDefault = DEFAULT,
         timeout: float | None = None,
         metadata: Sequence[tuple[str, str]] = (),
         filter: str | None = None,

{apache_airflow_providers_google-10.13.0rc1 → apache_airflow_providers_google-10.13.0rc2}/airflow/providers/google/cloud/operators/cloud_run.py

@@ -264,7 +264,7 @@ class CloudRunExecuteJobOperator(GoogleCloudBaseOperator):
     :param deferrable: Run operator in the deferrable mode
     """
 
-    template_fields = ("project_id", "region", "gcp_conn_id", "impersonation_chain", "job_name")
+    template_fields = ("project_id", "region", "gcp_conn_id", "impersonation_chain", "job_name", "overrides")
 
     def __init__(
         self,