fix: column_name resolution when parsing dbt generic tests (#798)

fal-ai · Mar 28, 2023 · f14e05b · f14e05b
1 parent 2b9247e
commit f14e05b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/projects/fal/src/faldbt/project.py b/projects/fal/src/faldbt/project.py
@@ -132,6 +132,10 @@ def __post_init__(self):
         assert isinstance(self.node, GenericTestNode)
         self.column = self.node.column_name
 
+        # Column name might be stored in test_metadata
+        if not self.column and self.node.test_metadata.kwargs.get("column_name"):
+            self.column = self.node.test_metadata.kwargs.get("column_name")
+
     @property
     def source_id(self):
         if self.source_ids: