openlineage-python 1.16.0__tar.gz → 1.17.1__tar.gz

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openlineage-python
-Version: 1.16.0
+Version: 1.17.1
 Summary: OpenLineage Python Client
 Author-email: OpenLineage <info@openlineage.io>
 Keywords: openlineage

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/constants.py

@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 from __future__ import annotations
 
-__version__ = "1.16.0"
+__version__ = "1.17.1"
 
 DEFAULT_TIMEOUT_MS = 5000
 DEFAULT_NAMESPACE_NAME = "default"

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/base.py

@@ -6,7 +6,7 @@ from __future__ import annotations
 from enum import Enum
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.constants import DEFAULT_PRODUCER
 from openlineage.client.utils import RedactMixin
 
@@ -18,13 +18,13 @@ def set_producer(producer: str) -> None:
     PRODUCER = producer
 
 
-@define(kw_only=True)
+@attr.define(kw_only=True)
 class BaseEvent(RedactMixin):
-    eventTime: str = field()  # noqa: N815
+    eventTime: str = attr.field()  # noqa: N815
     """the time the event occurred at"""
 
-    producer: str = field(default="", kw_only=True)
-    schemaURL: str = field(  # noqa: N815
+    producer: str = attr.field(default="", kw_only=True)
+    schemaURL: str = attr.field(  # noqa: N815
         default="https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/BaseEvent", init=False
     )
     _base_skip_redact: ClassVar[list[str]] = ["producer", "schemaURL"]
@@ -66,12 +66,12 @@ class BaseEvent(RedactMixin):
         urlparse(value)
 
 
-@define
+@attr.define
 class BaseFacet(RedactMixin):
     """all fields of the base facet are prefixed with _ to avoid name conflicts in facets"""
 
-    _producer: str = field(default="", kw_only=True)
-    _schemaURL: str = field(  # noqa: N815
+    _producer: str = attr.field(default="", kw_only=True)
+    _schemaURL: str = attr.field(  # noqa: N815
         default="https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/BaseFacet", init=False
     )
     _base_skip_redact: ClassVar[list[str]] = ["_producer", "_schemaURL"]
@@ -103,7 +103,7 @@ class BaseFacet(RedactMixin):
         urlparse(value)
 
 
-@define
+@attr.define
 class Dataset(RedactMixin):
     namespace: str
     """The namespace containing that dataset"""
@@ -111,7 +111,7 @@ class Dataset(RedactMixin):
     name: str
     """The unique name for that dataset within that namespace"""
 
-    facets: dict[str, DatasetFacet] | None = field(factory=dict, kw_only=True)  # type: ignore[assignment]
+    facets: dict[str, DatasetFacet] | None = attr.field(factory=dict, kw_only=True)
     """The facets for this dataset"""
 
     _skip_redact: ClassVar[list[str]] = ["namespace", "name"]
@@ -121,7 +121,7 @@ class Dataset(RedactMixin):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/Dataset"
 
 
-@define(kw_only=True)
+@attr.define(kw_only=True)
 class DatasetEvent(BaseEvent):
     dataset: StaticDataset
 
@@ -130,11 +130,11 @@ class DatasetEvent(BaseEvent):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/DatasetEvent"
 
 
-@define
+@attr.define
 class DatasetFacet(BaseFacet):
     """A Dataset Facet"""
 
-    _deleted: bool | None = field(default=None, kw_only=True)
+    _deleted: bool | None = attr.field(default=None, kw_only=True)
     """set to true to delete a facet"""
 
     @staticmethod
@@ -157,11 +157,11 @@ class EventType(Enum):
     OTHER = "OTHER"
 
 
-@define
+@attr.define
 class InputDataset(Dataset):
     """An input dataset"""
 
-    inputFacets: dict[str, InputDatasetFacet] | None = field(factory=dict)  # type: ignore[assignment]# noqa: N815
+    inputFacets: dict[str, InputDatasetFacet] | None = attr.field(factory=dict)  # noqa: N815
     """The input facets for this dataset."""
 
     _additional_skip_redact: ClassVar[list[str]] = ["namespace", "name"]
@@ -171,7 +171,7 @@ class InputDataset(Dataset):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/InputDataset"
 
 
-@define
+@attr.define
 class InputDatasetFacet(BaseFacet):
     """An Input Dataset Facet"""
 
@@ -180,7 +180,7 @@ class InputDatasetFacet(BaseFacet):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/InputDatasetFacet"
 
 
-@define
+@attr.define
 class Job(RedactMixin):
     namespace: str
     """The namespace containing that job"""
@@ -188,7 +188,7 @@ class Job(RedactMixin):
     name: str
     """The unique name for that job within that namespace"""
 
-    facets: dict[str, JobFacet] | None = field(factory=dict)  # type: ignore[assignment]
+    facets: dict[str, JobFacet] | None = attr.field(factory=dict)
     """The job facets."""
 
     _skip_redact: ClassVar[list[str]] = ["namespace", "name"]
@@ -198,13 +198,13 @@ class Job(RedactMixin):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/Job"
 
 
-@define(kw_only=True)
+@attr.define(kw_only=True)
 class JobEvent(BaseEvent):
     job: Job
-    inputs: list[InputDataset] | None = field(factory=list)  # type: ignore[assignment]
+    inputs: list[InputDataset] | None = attr.field(factory=list)
     """The set of **input** datasets."""
 
-    outputs: list[OutputDataset] | None = field(factory=list)  # type: ignore[assignment]
+    outputs: list[OutputDataset] | None = attr.field(factory=list)
     """The set of **output** datasets."""
 
     @staticmethod
@@ -212,11 +212,11 @@ class JobEvent(BaseEvent):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/JobEvent"
 
 
-@define
+@attr.define
 class JobFacet(BaseFacet):
     """A Job Facet"""
 
-    _deleted: bool | None = field(default=None, kw_only=True)
+    _deleted: bool | None = attr.field(default=None, kw_only=True)
     """set to true to delete a facet"""
 
     @staticmethod
@@ -224,11 +224,11 @@ class JobFacet(BaseFacet):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/JobFacet"
 
 
-@define
+@attr.define
 class OutputDataset(Dataset):
     """An output dataset"""
 
-    outputFacets: dict[str, OutputDatasetFacet] | None = field(factory=dict)  # type: ignore[assignment]# noqa: N815
+    outputFacets: dict[str, OutputDatasetFacet] | None = attr.field(factory=dict)  # noqa: N815
     """The output facets for this dataset"""
 
     _additional_skip_redact: ClassVar[list[str]] = ["namespace", "name"]
@@ -238,7 +238,7 @@ class OutputDataset(Dataset):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/OutputDataset"
 
 
-@define
+@attr.define
 class OutputDatasetFacet(BaseFacet):
     """An Output Dataset Facet"""
 
@@ -247,12 +247,12 @@ class OutputDatasetFacet(BaseFacet):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/OutputDatasetFacet"
 
 
-@define
+@attr.define
 class Run(RedactMixin):
-    runId: str = field()  # noqa: N815
+    runId: str = attr.field()  # noqa: N815
     """The globally unique ID of the run associated with the job."""
 
-    facets: dict[str, RunFacet] | None = field(factory=dict)  # type: ignore[assignment]
+    facets: dict[str, RunFacet] | None = attr.field(factory=dict)
     """The run facets."""
 
     _skip_redact: ClassVar[list[str]] = ["runId"]
@@ -268,20 +268,20 @@ class Run(RedactMixin):
         UUID(value)
 
 
-@define(kw_only=True)
+@attr.define(kw_only=True)
 class RunEvent(BaseEvent):
     run: Run
     job: Job
-    eventType: EventType | None = field(default=None)  # noqa: N815
+    eventType: EventType | None = attr.field(default=None)  # noqa: N815
     """
     the current transition of the run state. It is required to issue 1 START event and 1 of [ COMPLETE,
     ABORT, FAIL ] event per run. Additional events with OTHER eventType can be added to the same run.
     For example to send additional metadata after the run is complete
     """
-    inputs: list[InputDataset] | None = field(factory=list)  # type: ignore[assignment]
+    inputs: list[InputDataset] | None = attr.field(factory=list)
     """The set of **input** datasets."""
 
-    outputs: list[OutputDataset] | None = field(factory=list)  # type: ignore[assignment]
+    outputs: list[OutputDataset] | None = attr.field(factory=list)
     """The set of **output** datasets."""
 
     _additional_skip_redact: ClassVar[list[str]] = ["eventType", "eventTime"]
@@ -291,7 +291,7 @@ class RunEvent(BaseEvent):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/RunEvent"
 
 
-@define
+@attr.define
 class RunFacet(BaseFacet):
     """A Run Facet"""
 
@@ -300,7 +300,7 @@ class RunFacet(BaseFacet):
         return "https://openlineage.io/spec/2-0-2/OpenLineage.json#/$defs/RunFacet"
 
 
-@define
+@attr.define
 class StaticDataset(Dataset):
     """A Dataset sent within static metadata events"""
 

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/column_lineage_dataset.py

@@ -5,35 +5,35 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class ColumnLineageDatasetFacet(DatasetFacet):
     fields: dict[str, Fields]
     """Column level lineage that maps output fields into input fields used to evaluate them."""
 
     @staticmethod
     def _get_schema() -> str:
-        return "https://openlineage.io/spec/facets/1-0-2/ColumnLineageDatasetFacet.json#/$defs/ColumnLineageDatasetFacet"
+        return "https://openlineage.io/spec/facets/1-1-0/ColumnLineageDatasetFacet.json#/$defs/ColumnLineageDatasetFacet"
 
 
-@define
+@attr.define
 class Fields(RedactMixin):
     inputFields: list[InputField]  # noqa: N815
-    transformationDescription: str | None = field(default=None)  # noqa: N815
+    transformationDescription: str | None = attr.field(default=None)  # noqa: N815
     """a string representation of the transformation applied"""
 
-    transformationType: str | None = field(default=None)  # noqa: N815
+    transformationType: str | None = attr.field(default=None)  # noqa: N815
     """
     IDENTITY|MASKED reflects a clearly defined behavior. IDENTITY: exact same as input; MASKED: no
     original data available (like a hash of PII for example)
     """
 
 
-@define
+@attr.define
 class InputField(RedactMixin):
     namespace: str
     """The input dataset namespace"""
@@ -44,4 +44,22 @@ class InputField(RedactMixin):
     field: str
     """The input field"""
 
+    transformations: list[Transformation] | None = attr.field(factory=list)
     _skip_redact: ClassVar[list[str]] = ["namespace", "name", "field"]
+
+
+@attr.define
+class Transformation(RedactMixin):
+    type: str
+    """The type of the transformation. Allowed values are: DIRECT, INDIRECT"""
+
+    subtype: str | None = attr.field(default=None)
+    """The subtype of the transformation"""
+
+    description: str | None = attr.field(default=None)
+    """a string representation of the transformation applied"""
+
+    masking: bool | None = attr.field(default=None)
+    """is transformation masking the data or not"""
+
+    _skip_redact: ClassVar[list[str]] = ["type", "subtype", "masking"]

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/data_quality_assertions_dataset.py

@@ -5,18 +5,18 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import InputDatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Assertion(RedactMixin):
     assertion: str
     """Type of expectation test that dataset is subjected to"""
 
     success: bool
-    column: str | None = field(default=None)
+    column: str | None = attr.field(default=None)
     """
     Column that expectation is testing. It should match the name provided in SchemaDatasetFacet. If
     column field is empty, then expectation refers to whole dataset.
@@ -24,7 +24,7 @@ class Assertion(RedactMixin):
     _skip_redact: ClassVar[list[str]] = ["column"]
 
 
-@define
+@attr.define
 class DataQualityAssertionsDatasetFacet(InputDatasetFacet):
     """list of tests performed on dataset or dataset columns, and their results"""
 

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/data_quality_metrics_input_dataset.py

@@ -3,43 +3,43 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import InputDatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class ColumnMetrics(RedactMixin):
-    nullCount: int | None = field(default=None)  # noqa: N815
+    nullCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of null values in this column for the rows evaluated"""
 
-    distinctCount: int | None = field(default=None)  # noqa: N815
+    distinctCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of distinct values in this column for the rows evaluated"""
 
-    sum: float | None = field(default=None)
+    sum: float | None = attr.field(default=None)
     """The total sum of values in this column for the rows evaluated"""
 
-    count: float | None = field(default=None)
+    count: float | None = attr.field(default=None)
     """The number of values in this column"""
 
-    min: float | None = field(default=None)
-    max: float | None = field(default=None)
-    quantiles: dict[str, float] | None = field(factory=dict)  # type: ignore[assignment]
+    min: float | None = attr.field(default=None)
+    max: float | None = attr.field(default=None)
+    quantiles: dict[str, float] | None = attr.field(factory=dict)
     """The property key is the quantile. Examples: 0.1 0.25 0.5 0.75 1"""
 
 
-@define
+@attr.define
 class DataQualityMetricsInputDatasetFacet(InputDatasetFacet):
     columnMetrics: dict[str, ColumnMetrics]  # noqa: N815
     """The property key is the column name"""
 
-    rowCount: int | None = field(default=None)  # noqa: N815
+    rowCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of rows evaluated"""
 
-    bytes: int | None = field(default=None)
+    bytes: int | None = attr.field(default=None)
     """The size in bytes"""
 
-    fileCount: int | None = field(default=None)  # noqa: N815
+    fileCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of files evaluated"""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/dataset_version_dataset.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define
+import attr
 from openlineage.client.generated.base import DatasetFacet
 
 
-@define
+@attr.define
 class DatasetVersionDatasetFacet(DatasetFacet):
     datasetVersion: str  # noqa: N815
     """The version of the dataset."""

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/datasource_dataset.py

@@ -5,14 +5,14 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 
 
-@define
+@attr.define
 class DatasourceDatasetFacet(DatasetFacet):
-    name: str | None = field(default=None)
-    uri: str | None = field(default=None)
+    name: str | None = attr.field(default=None)
+    uri: str | None = attr.field(default=None)
     _additional_skip_redact: ClassVar[list[str]] = ["name", "uri"]
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/documentation_dataset.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define
+import attr
 from openlineage.client.generated.base import DatasetFacet
 
 
-@define
+@attr.define
 class DocumentationDatasetFacet(DatasetFacet):
     description: str
     """The description of the dataset."""

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/documentation_job.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define
+import attr
 from openlineage.client.generated.base import JobFacet
 
 
-@define
+@attr.define
 class DocumentationJobFacet(JobFacet):
     description: str
     """The description of the job."""

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/error_message_run.py

@@ -5,11 +5,11 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import RunFacet
 
 
-@define
+@attr.define
 class ErrorMessageRunFacet(RunFacet):
     message: str
     """A human-readable string representing error message generated by observed system"""
@@ -17,7 +17,7 @@ class ErrorMessageRunFacet(RunFacet):
     programmingLanguage: str  # noqa: N815
     """Programming language the observed system uses."""
 
-    stackTrace: str | None = field(default=None)  # noqa: N815
+    stackTrace: str | None = attr.field(default=None)  # noqa: N815
     """A language-specific stack trace generated by observed system"""
 
     _additional_skip_redact: ClassVar[list[str]] = ["programmingLanguage"]

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/external_query_run.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define
+import attr
 from openlineage.client.generated.base import RunFacet
 
 
-@define
+@attr.define
 class ExternalQueryRunFacet(RunFacet):
     externalQueryId: str  # noqa: N815
     """Identifier for the external system"""

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/extraction_error_run.py

@@ -3,29 +3,29 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import RunFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Error(RedactMixin):
     errorMessage: str  # noqa: N815
     """Text representation of extraction error message."""
 
-    stackTrace: str | None = field(default=None)  # noqa: N815
+    stackTrace: str | None = attr.field(default=None)  # noqa: N815
     """Stack trace of extraction error message"""
 
-    task: str | None = field(default=None)
+    task: str | None = attr.field(default=None)
     """
     Text representation of task that failed. This can be, for example, SQL statement that parser could
     not interpret.
     """
-    taskNumber: int | None = field(default=None)  # noqa: N815
+    taskNumber: int | None = attr.field(default=None)  # noqa: N815
     """Order of task (counted from 0)."""
 
 
-@define
+@attr.define
 class ExtractionErrorRunFacet(RunFacet):
     totalTasks: int  # noqa: N815
     """

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/job_type_job.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import JobFacet
 
 
-@define
+@attr.define
 class JobTypeJobFacet(JobFacet):
     processingType: str  # noqa: N815
     """Job processing type like: BATCH or STREAMING"""
@@ -15,7 +15,7 @@ class JobTypeJobFacet(JobFacet):
     integration: str
     """OpenLineage integration type of this job: SPARK|DBT|AIRFLOW|FLINK"""
 
-    jobType: str | None = field(default=None)  # noqa: N815
+    jobType: str | None = attr.field(default=None)  # noqa: N815
     """Run type like: QUERY|COMMAND|DAG|TASK|JOB|MODEL"""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/lifecycle_state_change_dataset.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 
 from enum import Enum
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 from openlineage.client.utils import RedactMixin
 
@@ -21,12 +21,12 @@ class LifecycleStateChange(Enum):
     TRUNCATE = "TRUNCATE"
 
 
-@define
+@attr.define
 class LifecycleStateChangeDatasetFacet(DatasetFacet):
     lifecycleStateChange: LifecycleStateChange  # noqa: N815
     """The lifecycle state change."""
 
-    previousIdentifier: PreviousIdentifier | None = field(default=None)  # noqa: N815
+    previousIdentifier: PreviousIdentifier | None = attr.field(default=None)  # noqa: N815
     """Previous name of the dataset in case of renaming it."""
 
     @staticmethod
@@ -34,7 +34,7 @@ class LifecycleStateChangeDatasetFacet(DatasetFacet):
         return "https://openlineage.io/spec/facets/1-0-1/LifecycleStateChangeDatasetFacet.json#/$defs/LifecycleStateChangeDatasetFacet"
 
 
-@define
+@attr.define
 class PreviousIdentifier(RedactMixin):
     """Previous name of the dataset in case of renaming it."""
 

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/nominal_time_run.py

@@ -5,18 +5,18 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import RunFacet
 
 
-@define
+@attr.define
 class NominalTimeRunFacet(RunFacet):
-    nominalStartTime: str = field()  # noqa: N815
+    nominalStartTime: str = attr.field()  # noqa: N815
     """
     An [ISO-8601](https://en.wikipedia.org/wiki/ISO_8601) timestamp representing the nominal start time
     (included) of the run. AKA the schedule time
     """
-    nominalEndTime: str | None = field(default=None)  # noqa: N815
+    nominalEndTime: str | None = attr.field(default=None)  # noqa: N815
     """
     An [ISO-8601](https://en.wikipedia.org/wiki/ISO_8601) timestamp representing the nominal end time
     (excluded) of the run. (Should be the nominal start time of the next run)

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/output_statistics_output_dataset.py

@@ -5,19 +5,19 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import OutputDatasetFacet
 
 
-@define
+@attr.define
 class OutputStatisticsOutputDatasetFacet(OutputDatasetFacet):
-    rowCount: int | None = field(default=None)  # noqa: N815
+    rowCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of rows written to the dataset"""
 
-    size: int | None = field(default=None)
+    size: int | None = attr.field(default=None)
     """The size in bytes written to the dataset"""
 
-    fileCount: int | None = field(default=None)  # noqa: N815
+    fileCount: int | None = attr.field(default=None)  # noqa: N815
     """The number of files written to the dataset"""
 
     _additional_skip_redact: ClassVar[list[str]] = ["rowCount", "size"]

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/ownership_dataset.py

@@ -3,25 +3,25 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Owner(RedactMixin):
     name: str
     """
     the identifier of the owner of the Dataset. It is recommended to define this as a URN. For example
     application:foo, user:jdoe, team:data
     """
-    type: str | None = field(default=None)
+    type: str | None = attr.field(default=None)
     """The type of ownership (optional)"""
 
 
-@define
+@attr.define
 class OwnershipDatasetFacet(DatasetFacet):
-    owners: list[Owner] | None = field(factory=list)  # type: ignore[assignment]
+    owners: list[Owner] | None = attr.field(factory=list)
     """The owners of the dataset."""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/ownership_job.py

@@ -3,25 +3,25 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import JobFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Owner(RedactMixin):
     name: str
     """
     the identifier of the owner of the Job. It is recommended to define this as a URN. For example
     application:foo, user:jdoe, team:data
     """
-    type: str | None = field(default=None)
+    type: str | None = attr.field(default=None)
     """The type of ownership (optional)"""
 
 
-@define
+@attr.define
 class OwnershipJobFacet(JobFacet):
-    owners: list[Owner] | None = field(factory=list)  # type: ignore[assignment]
+    owners: list[Owner] | None = attr.field(factory=list)
     """The owners of the job."""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/parent_run.py

@@ -5,12 +5,12 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import RunFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Job(RedactMixin):
     namespace: str
     """The namespace containing that job"""
@@ -19,7 +19,7 @@ class Job(RedactMixin):
     """The unique name for that job within that namespace"""
 
 
-@define
+@attr.define
 class ParentRunFacet(RunFacet):
     """
     the id of the parent run and job, iff this run was spawn from an other run (for example, the Dag run
@@ -46,9 +46,9 @@ class ParentRunFacet(RunFacet):
         return cls(run=Run(runId=runId), job=Job(namespace=namespace, name=name))
 
 
-@define
+@attr.define
 class Run(RedactMixin):
-    runId: str = field()  # noqa: N815
+    runId: str = attr.field()  # noqa: N815
     """The globally unique ID of the run associated with the job."""
 
     @runId.validator

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/processing_engine_run.py

@@ -3,19 +3,19 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import RunFacet
 
 
-@define
+@attr.define
 class ProcessingEngineRunFacet(RunFacet):
     version: str
     """Processing engine version. Might be Airflow or Spark version."""
 
-    name: str | None = field(default=None)
+    name: str | None = attr.field(default=None)
     """Processing engine name, e.g. Airflow or Spark"""
 
-    openlineageAdapterVersion: str | None = field(default=None)  # noqa: N815
+    openlineageAdapterVersion: str | None = attr.field(default=None)  # noqa: N815
     """OpenLineage adapter package version. Might be e.g. OpenLineage Airflow integration package version"""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/schema_dataset.py

@@ -3,14 +3,14 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class SchemaDatasetFacet(DatasetFacet):
-    fields: list[SchemaDatasetFacetFields] | None = field(factory=list)  # type: ignore[assignment]
+    fields: list[SchemaDatasetFacetFields] | None = attr.field(factory=list)
     """The fields of the data source."""
 
     @staticmethod
@@ -18,18 +18,18 @@ class SchemaDatasetFacet(DatasetFacet):
         return "https://openlineage.io/spec/facets/1-1-1/SchemaDatasetFacet.json#/$defs/SchemaDatasetFacet"
 
 
-@define
+@attr.define
 class SchemaDatasetFacetFields(RedactMixin):
     name: str
     """The name of the field."""
 
-    type: str | None = field(default=None)
+    type: str | None = attr.field(default=None)
     """The type of the field."""
 
-    description: str | None = field(default=None)
+    description: str | None = attr.field(default=None)
     """The description of the field."""
 
-    fields: list[SchemaDatasetFacetFields] | None = field(factory=list)  # type: ignore[assignment]
+    fields: list[SchemaDatasetFacetFields] | None = attr.field(factory=list)
     """Nested struct fields."""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/source_code_job.py

@@ -5,11 +5,11 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define
+import attr
 from openlineage.client.generated.base import JobFacet
 
 
-@define
+@attr.define
 class SourceCodeJobFacet(JobFacet):
     language: str
     """Language in which source code of this job was written."""

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/source_code_location_job.py

@@ -5,31 +5,31 @@ from __future__ import annotations
 
 from typing import ClassVar
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import JobFacet
 
 
-@define
+@attr.define
 class SourceCodeLocationJobFacet(JobFacet):
     type: str
     """the source control system"""
 
-    url: str = field()
+    url: str = attr.field()
     """the full http URL to locate the file"""
 
-    repoUrl: str | None = field(default=None)  # noqa: N815
+    repoUrl: str | None = attr.field(default=None)  # noqa: N815
     """the URL to the repository"""
 
-    path: str | None = field(default=None)
+    path: str | None = attr.field(default=None)
     """the path in the repo containing the source files"""
 
-    version: str | None = field(default=None)
+    version: str | None = attr.field(default=None)
     """the current version deployed (not a branch name, the actual unique version)"""
 
-    tag: str | None = field(default=None)
+    tag: str | None = attr.field(default=None)
     """optional tag name"""
 
-    branch: str | None = field(default=None)
+    branch: str | None = attr.field(default=None)
     """optional branch name"""
 
     _additional_skip_redact: ClassVar[list[str]] = ["type", "url"]

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/sql_job.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from attr import define
+import attr
 from openlineage.client.generated.base import JobFacet
 
 
-@define
+@attr.define
 class SQLJobFacet(JobFacet):
     query: str
 

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/storage_dataset.py

@@ -3,16 +3,16 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 
 
-@define
+@attr.define
 class StorageDatasetFacet(DatasetFacet):
     storageLayer: str  # noqa: N815
     """Storage layer provider with allowed values: iceberg, delta."""
 
-    fileFormat: str | None = field(default=None)  # noqa: N815
+    fileFormat: str | None = attr.field(default=None)  # noqa: N815
     """File format with allowed values: parquet, orc, avro, json, csv, text, xml."""
 
     @staticmethod

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/generated/symlinks_dataset.py

@@ -3,12 +3,12 @@
 
 from __future__ import annotations
 
-from attr import define, field
+import attr
 from openlineage.client.generated.base import DatasetFacet
 from openlineage.client.utils import RedactMixin
 
 
-@define
+@attr.define
 class Identifier(RedactMixin):
     namespace: str
     """The dataset namespace"""
@@ -20,9 +20,9 @@ class Identifier(RedactMixin):
     """Identifier type"""
 
 
-@define
+@attr.define
 class SymlinksDatasetFacet(DatasetFacet):
-    identifiers: list[Identifier] | None = field(factory=list)  # type: ignore[assignment]
+    identifiers: list[Identifier] | None = attr.field(factory=list)
 
     @staticmethod
     def _get_schema() -> str:

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/run.py

@@ -76,8 +76,8 @@ class JobEvent(RedactMixin):
     producer: str = attr.ib()
     schemaURL: str = attr.ib()  # noqa: N815
     job: Job = attr.ib()
-    inputs: Optional[List[Dataset]] = attr.ib(factory=list)  # type: ignore[assignment]
-    outputs: Optional[List[Dataset]] = attr.ib(factory=list)  # type: ignore[assignment]
+    inputs: Optional[List[Dataset]] = attr.ib(factory=list)
+    outputs: Optional[List[Dataset]] = attr.ib(factory=list)
 
     _skip_redact: ClassVar[List[str]] = ["producer"]
 
@@ -101,8 +101,8 @@ class RunEvent(RedactMixin):
     run: Run = attr.ib()
     job: Job = attr.ib()
     producer: str = attr.ib()
-    inputs: Optional[List[Dataset]] = attr.ib(factory=list)  # type: ignore[assignment]
-    outputs: Optional[List[Dataset]] = attr.ib(factory=list)  # type: ignore[assignment]
+    inputs: Optional[List[Dataset]] = attr.ib(factory=list)
+    outputs: Optional[List[Dataset]] = attr.ib(factory=list)
     schemaURL: str = attr.ib(default=SCHEMA_URL)  # noqa: N815
 
     _skip_redact: ClassVar[List[str]] = ["eventType", "eventTime", "producer", "schemaURL"]

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/transport/kafka.py

@@ -164,7 +164,7 @@ class KafkaTransport(Transport):
 
 def _check_if_airflow_sqlalchemy_context() -> bool:
     try:
-        from airflow.version import version  # type: ignore[import]
+        from airflow.version import version  # type: ignore[import-not-found]
 
         parsed_version = Version(version)
         if Version("2.3.0") <= parsed_version < Version("2.6.0"):

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/openlineage/client/transport/msk_iam.py

@@ -15,7 +15,7 @@ log = logging.getLogger(__name__)
 
 def _detect_running_region() -> None | str:
     """Dynamically determine the region."""
-    import boto3  # type: ignore[import]
+    import boto3  # type: ignore[import-untyped]
 
     checks = [
         # check if set through ENV vars
@@ -44,7 +44,7 @@ class MSKIAMConfig(KafkaConfig):
 
 
 def _oauth_cb(config: MSKIAMConfig, *_: Any) -> tuple[str, float]:
-    from aws_msk_iam_sasl_signer import MSKAuthTokenProvider  # type: ignore[import]
+    from aws_msk_iam_sasl_signer import MSKAuthTokenProvider  # type: ignore[import-untyped]
 
     region = config.region
     if config.aws_profile:

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/pyproject.toml

@@ -6,7 +6,7 @@ requires = [
 
 [project]
 name = "openlineage-python"
-version = "1.16.0"
+version = "1.17.1"
 description = "OpenLineage Python Client"
 readme = "README.md"
 keywords = [

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/redact_fields.yml

@@ -56,6 +56,11 @@
         - namespace
         - name
         - field
+    - class_name: Transformation
+      redact_fields:
+        - type
+        - subtype
+        - masking
 - module: data_quality_assertions_dataset
   classes:
     - class_name: Assertion

{openlineage_python-1.16.0 → openlineage_python-1.17.1}/tox.ini

@@ -14,6 +14,8 @@ skip_missing_interpreters = true
 description = run the unit tests with pytest under {basepython}
 package = wheel
 wheel_build_env = .pkg
+allowlist_externals = uv
+install_command = uv pip install {opts} {packages}
 extras =
     kafka
     msk-iam
@@ -33,7 +35,7 @@ commands =
 [testenv:type]
 description = run type check on code base
 deps =
-    mypy==1.3
+    mypy==1.10
     numpy==1.24.3
     types-PyYAML==6.0.12.10
     types-python-dateutil==2.8.19.13