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 e277d9f0c6..2ca8fe3ec3 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 @@ -2,7 +2,6 @@ import logging from collections import defaultdict from dataclasses import dataclass, field from datetime import datetime, timezone -from enum import Enum from typing import Any, Dict, List, Optional, cast from google.cloud import bigquery @@ -19,7 +18,7 @@ from datahub.ingestion.source.sql.sql_generic import BaseColumn, BaseTable, Base logger: logging.Logger = logging.getLogger(__name__) -class BigqueryTableType(Enum): +class BigqueryTableType: # See https://cloud.google.com/bigquery/docs/information-schema-tables#schema BASE_TABLE = "BASE TABLE" EXTERNAL = "EXTERNAL" diff --git a/metadata-ingestion/tests/unit/test_bigquery_source.py b/metadata-ingestion/tests/unit/test_bigquery_source.py index d464af5d10..02e2d8a580 100644 --- a/metadata-ingestion/tests/unit/test_bigquery_source.py +++ b/metadata-ingestion/tests/unit/test_bigquery_source.py @@ -18,7 +18,6 @@ from datahub.ingestion.source.bigquery_v2.bigquery_config import BigQueryV2Confi from datahub.ingestion.source.bigquery_v2.bigquery_schema import ( BigQueryDataDictionary, BigqueryProject, - BigqueryTableType, BigqueryView, ) from datahub.ingestion.source.bigquery_v2.lineage import LineageEdge @@ -464,7 +463,7 @@ def test_get_views_for_dataset( last_altered=bigquery_view_1.last_altered, comment=bigquery_view_1.comment, view_definition=bigquery_view_1.view_definition, - table_type=BigqueryTableType.VIEW, + table_type="VIEW", ) ) row2 = create_row( # Materialized view, no last_altered @@ -473,7 +472,7 @@ def test_get_views_for_dataset( created=bigquery_view_2.created, comment=bigquery_view_2.comment, view_definition=bigquery_view_2.view_definition, - table_type=BigqueryTableType.MATERIALIZED_VIEW, + table_type="MATERIALIZED VIEW", ) ) query_mock.return_value = [row1, row2]