diff --git a/ingestion/src/metadata/profiler/processor/metric_filter.py b/ingestion/src/metadata/profiler/processor/metric_filter.py index a4b2ea8c3e3..d2a1a5a3f6d 100644 --- a/ingestion/src/metadata/profiler/processor/metric_filter.py +++ b/ingestion/src/metadata/profiler/processor/metric_filter.py @@ -168,7 +168,9 @@ class MetricFilter: if not isinstance(column, SQALikeColumn): mapper = converter_registry[service_type] sqa_to_om_types = mapper.map_sqa_to_om_types() - om_data_types: Optional[Set] = sqa_to_om_types.get(column.type, None) + om_data_types: Optional[Set] = sqa_to_om_types.get( + column.type.__class__, None + ) else: om_data_types = {column.type} diff --git a/ingestion/tests/integration/profiler/test_sqa_profiler.py b/ingestion/tests/integration/profiler/test_sqa_profiler.py index c8fb6771de5..2d45da3f775 100644 --- a/ingestion/tests/integration/profiler/test_sqa_profiler.py +++ b/ingestion/tests/integration/profiler/test_sqa_profiler.py @@ -112,7 +112,7 @@ class TestSQAProfiler(TestCase): tables: List[Table] = self.metadata.list_all_entities(Table) for table in tables: - if table.name != "users": + if table.name.__root__ != "users": continue table = self.metadata.get_latest_table_profile(table.fullyQualifiedName) columns = table.columns @@ -162,7 +162,7 @@ class TestSQAProfiler(TestCase): tables: List[Table] = self.metadata.list_all_entities(Table) for table in tables: - if table.name != "users": + if table.name.__root__ != "users": continue table = self.metadata.get_latest_table_profile(table.fullyQualifiedName) columns = table.columns