Update src/databricks/labs/ucx/hive_metastore/tables.py

nfx · web-flow · commit ebb3c411ccc9 · 2023-09-28T18:59:16.000+02:00
diff --git a/src/databricks/labs/ucx/hive_metastore/tables.py b/src/databricks/labs/ucx/hive_metastore/tables.py
@@ -174,10 +174,10 @@ def _init_default_catalog(default_catalog):
     def migrate_tables(self):
         tasks = []
         for table in self._tc.snapshot():
+            target_catalog = self._default_catalog
             if self._database_to_catalog_mapping:
-                tasks.append(partial(self._migrate_table, self._database_to_catalog_mapping[table.database], table))
-            else:
-                tasks.append(partial(self._migrate_table, self._default_catalog, table))
+                target_catalog = self._database_to_catalog_mapping[table.database]
+            tasks.append(partial(self._migrate_table, target_catalog, table))
         ThreadedExecution.gather("migrate tables", tasks)
 
     def _migrate_table(self, target_catalog, table):
