diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py index dc9dc97c9b0ea..a768a8a474927 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py +++ b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py @@ -165,6 +165,7 @@ def _build_source_map(self) -> Dict[str, DremioSourceMapping]: ), re.IGNORECASE ): + source_platform_name = source_platform_name.lower() source_map[source_platform_name] = mapping source_map[source_platform_name].dremio_source_type = ( self.reference_source_mapping.get_category( @@ -195,7 +196,7 @@ def _build_source_map(self) -> Dict[str, DremioSourceMapping]: source_type ) - source_map[source_platform_name] = DremioSourceMapping( + source_map[source_platform_name.lower()] = DremioSourceMapping( platform=source_type, platform_name=source_name, dremio_source_type=dremio_source_type, @@ -585,7 +586,7 @@ def _map_dremio_dataset_to_urn( Map a Dremio dataset to a DataHub URN. """ - mapping = self.source_map.get(dremio_source) + mapping = self.source_map.get(dremio_source.lower()) if not mapping: return None @@ -603,14 +604,14 @@ def _map_dremio_dataset_to_urn( if mapping.platform_instance: return make_dataset_urn_with_platform_instance( - platform=mapping.platform, + platform=mapping.platform.lower(), name=dremio_dataset, platform_instance=mapping.platform_instance, env=self.config.env ) return make_dataset_urn_with_platform_instance( - platform=mapping.platform, + platform=mapping.platform.lower(), name=dremio_dataset, platform_instance=None, env=self.config.env