Skip to content

Commit 6df3ba8

Browse files
Get test_metadata optionally everywhere
1 parent 05d7bf3 commit 6df3ba8

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/dbt_score/models.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,10 @@ def _get_columns(
141141
[
142142
test
143143
for test in test_values
144-
if test["test_metadata"]["kwargs"].get("column_name") == name
144+
if test.get("test_metadata", {})
145+
.get("kwargs", {})
146+
.get("column_name")
147+
== name
145148
],
146149
)
147150
for name, values in node_values.get("columns", {}).items()
@@ -224,7 +227,9 @@ def from_node(
224227
tests=[
225228
Test.from_node(test)
226229
for test in test_values
227-
if not test["test_metadata"]["kwargs"].get("column_name")
230+
if not test.get("test_metadata", {})
231+
.get("kwargs", {})
232+
.get("column_name")
228233
],
229234
depends_on=node_values["depends_on"],
230235
_raw_values=node_values,

0 commit comments

Comments
 (0)