Fixed Loop

Author: FastLee

src/databricks/labs/ucx/hive_metastore/table_migration_status.py

@@ -116,23 +116,22 @@ def get_seen_tables(self, *, scope: set[Table] | None = None) -> dict[str, str]:
                     schema.name,
                 )
             )
-            tables: list = []
-            table_lists = Threads.gather("list tables", tasks, self.API_LIMIT)
-            # Combine tuple of lists to a list
-            for table_list in table_lists[0]:
-                if table_list is not None:
-                    tables.extend(table_list)
-            for table in tables:
-                if not isinstance(table, TableInfo):
-                    logger.warning(f"Table {table} is not an instance of TableInfo")
-                    continue
-                if not table.full_name:
-                    logger.warning(f"The table {table.name} in {schema.name} has no full name")
-                    continue
-                if table_scope and table.full_name.lower() not in table_scope:
-                    continue
-                if not table.properties or "upgraded_from" not in table.properties:
-                    continue
+        tables: list = []
+        table_lists = Threads.gather("list tables", tasks, self.API_LIMIT)
+        # Combine tuple of lists to a list
+        for table_list in table_lists:
+            tables.extend(table_list)
+        for table in tables:
+            if not isinstance(table, TableInfo):
+                logger.warning(f"Table {table} is not an instance of TableInfo")
+                continue
+            if not table.full_name:
+                logger.warning(f"The table {table.name} in {schema.name} has no full name")
+                continue
+            if table_scope and table.full_name.lower() not in table_scope:
+                continue
+            if not table.properties or "upgraded_from" not in table.properties:
+                continue
 
                 seen_tables[table.full_name.lower()] = table.properties["upgraded_from"].lower()
         return seen_tables