Update sql_common.py

datahub-project · Dec 2, 2024 · 0f25c53 · 0f25c53
1 parent e2f6b26
commit 0f25c53
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py b/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py
@@ -786,11 +786,13 @@ def _process_table(
                 entityUrn=dataset_snapshot.urn,
                 aspect=UpstreamLineage(
                     upstreams=[external_upstream_table],
-                    fineGrainedLineages=self.get_fine_grained_lineages(
-                        dataset_urn=dataset_snapshot.urn,
-                        upstream_dataset_urn=location_urn,
-                        schema_fields=schema_fields,
-                    ),
+                    fineGrainedLineages=(
+                        self.get_fine_grained_lineages(
+                            dataset_urn=dataset_snapshot.urn,
+                            upstream_dataset_urn=location_urn,
+                            schema_fields=schema_fields,
+                        ),
+                    ) if self.ctx.graph else None
                 ),
             ).as_workunit()