fix: only set tags on columns that exist

remoteoss · Sep 25, 2024 · d12836e · d12836e
1 parent a53c82c
commit d12836e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/snowflake_utils/models/table.py b/snowflake_utils/models/table.py
@@ -394,6 +394,7 @@ def sync_tags(self, cursor: SnowflakeCursor) -> None:
 
     def sync_tags_columns(self, cursor: SnowflakeCursor) -> None:
         tags = self.current_column_tags()
+        columns = [c.name.upper() for c in self.get_columns(cursor)]
         existing_tags = {
             f"{column}.{tag_name}.{tags[column][tag_name]}".casefold(): (
                 column,
@@ -410,6 +411,7 @@ def sync_tags_columns(self, cursor: SnowflakeCursor) -> None:
             for tag_name, tag_value in self.table_structure.columns[
                 column.casefold()
             ].tags.items()
+            if column.upper() in columns
         }
 
         for tag in existing_tags: