From cb5ee34a1bf6aedd87d9fe822494c7b8e87f6cd2 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Sat, 22 Apr 2023 22:01:30 +0530
Subject: [PATCH] Fix Lineage Via Table Entity Error (#11209)

---
 .../metadata/ingestion/source/database/common_db_source.py  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ingestion/src/metadata/ingestion/source/database/common_db_source.py b/ingestion/src/metadata/ingestion/source/database/common_db_source.py
index 7e6872723a8..c9cc5062739 100644
--- a/ingestion/src/metadata/ingestion/source/database/common_db_source.py
+++ b/ingestion/src/metadata/ingestion/source/database/common_db_source.py
@@ -396,6 +396,9 @@ class CommonDbSourceService(
                 table_request.tableType = TableType.Partitioned.value
                 table_request.tablePartition = partition_details
 
+            yield table_request
+            self.register_record(table_request=table_request)
+
             if table_type == TableType.View or view_definition:
                 table_view = TableView.parse_obj(
                     {
@@ -407,9 +410,6 @@ class CommonDbSourceService(
                 )
                 self.context.table_views.append(table_view)
 
-            yield table_request
-            self.register_record(table_request=table_request)
-
             if table_constraints or foreign_columns:
                 self.context.table_constrains.append(
                     OMetaTableConstraints(