fix(ingest/redshift): Adding env parameter where it was missing for u…

…rn generation (#8199)
datahub-project · Jun 9, 2023 · 8d31a18 · 8d31a18
1 parent a04631b
commit 8d31a18
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/redshift/redshift.py b/metadata-ingestion/src/datahub/ingestion/source/redshift/redshift.py
@@ -618,6 +618,7 @@ def gen_view_dataset_workunits(
             platform=self.platform,
             name=datahub_dataset_name,
             platform_instance=self.config.platform_instance,
+            env=self.config.env,
         )
         if view.ddl:
             view_properties_aspect = ViewProperties(
@@ -697,6 +698,7 @@ def gen_dataset_workunits(
             platform=self.platform,
             name=datahub_dataset_name,
             platform_instance=self.config.platform_instance,
+            env=self.config.env,
         )
 
         yield from self.gen_schema_metadata(
@@ -919,6 +921,7 @@ def generate_lineage(self, database: str) -> Iterable[MetadataWorkUnit]:
                     platform=self.platform,
                     name=datahub_dataset_name,
                     platform_instance=self.config.platform_instance,
+                    env=self.config.env,
                 )
 
                 lineage_info = self.lineage_extractor.get_lineage(
@@ -935,9 +938,9 @@ def generate_lineage(self, database: str) -> Iterable[MetadataWorkUnit]:
             for view in self.db_views[database][schema]:
                 datahub_dataset_name = f"{database}.{schema}.{view.name}"
                 dataset_urn = make_dataset_urn_with_platform_instance(
-                    self.platform,
-                    datahub_dataset_name,
-                    self.config.platform_instance,
+                    platform=self.platform,
+                    name=datahub_dataset_name,
+                    platform_instance=self.config.platform_instance,
                     env=self.config.env,
                 )
                 lineage_info = self.lineage_extractor.get_lineage(