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 b911f4659c53f..fe221c56faab0 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 @@ 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 @@ 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 a890dbb4c9e37..b006fbc613697 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_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]