From 9f20a23e0028343eac574fab032607b9155df85a Mon Sep 17 00:00:00 2001 From: Mayuri Nehate <33225191+mayurinehate@users.noreply.github.com> Date: Mon, 9 Jan 2023 13:37:12 +0530 Subject: [PATCH] fix(ingest): unfreeze bigquery/snowflake column dataclass (#6921) --- .../src/datahub/ingestion/source/bigquery_v2/bigquery_schema.py | 2 +- .../src/datahub/ingestion/source/snowflake/snowflake_schema.py | 2 +- .../src/datahub/ingestion/source/sql/sql_generic.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_schema.py b/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_schema.py index 4a25e5e3fc..3bc8ecc985 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_schema.py +++ b/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_schema.py @@ -13,7 +13,7 @@ from datahub.ingestion.source.sql.sql_generic import BaseColumn, BaseTable, Base logger: logging.Logger = logging.getLogger(__name__) -@dataclass(frozen=True, eq=True) +@dataclass class BigqueryColumn(BaseColumn): field_path: str is_partition_column: bool diff --git a/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_schema.py b/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_schema.py index 47cef09b02..634ae6506b 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_schema.py +++ b/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_schema.py @@ -45,7 +45,7 @@ class SnowflakeTag: return f"{self.database}.{self.schema}.{self.name}" -@dataclass(frozen=True, eq=True) +@dataclass class SnowflakeColumn(BaseColumn): character_maximum_length: Optional[int] numeric_precision: Optional[int] diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/sql_generic.py b/metadata-ingestion/src/datahub/ingestion/source/sql/sql_generic.py index 5acbb1983f..3fbe8c40df 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/sql/sql_generic.py +++ b/metadata-ingestion/src/datahub/ingestion/source/sql/sql_generic.py @@ -16,7 +16,7 @@ from datahub.ingestion.api.decorators import ( from datahub.ingestion.source.sql.sql_common import SQLAlchemyConfig, SQLAlchemySource -@dataclass(frozen=True, eq=True) +@dataclass class BaseColumn: name: str ordinal_position: int