fix(mssql): use include_lineage flag from config

datahub-project · Jan 31, 2025 · e6d5639 · e6d5639
1 parent 301d628
commit e6d5639
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/mssql/source.py b/metadata-ingestion/src/datahub/ingestion/source/sql/mssql/source.py
@@ -466,7 +466,7 @@ def _process_stored_procedure(
         yield from self.construct_job_workunits(
             data_job,
             # For stored procedure lineage is ingested later
-            include_lineage=False,
+            override_include_lineage=False,
         )
 
     @staticmethod
@@ -627,7 +627,7 @@ def _get_stored_procedures(
     def construct_job_workunits(
         self,
         data_job: MSSQLDataJob,
-        include_lineage: bool = True,
+        override_include_lineage: Option[bool] = None,
     ) -> Iterable[MetadataWorkUnit]:
         yield MetadataChangeProposalWrapper(
             entityUrn=data_job.urn,
@@ -641,7 +641,12 @@ def construct_job_workunits(
                 aspect=data_platform_instance_aspect,
             ).as_workunit()
 
-        if include_lineage:
+        effective_include_lineage = (
+            override_include_lineage
+            if override_include_lineage is not None
+            else self.config.include_lineage
+        )
+        if effective_include_lineage:
             yield MetadataChangeProposalWrapper(
                 entityUrn=data_job.urn,
                 aspect=data_job.as_datajob_input_output_aspect,