Rearrange construction of params for run_model tracking

dbt-labs · Apr 11, 2023 · 44440ba · 44440ba
1 parent aae96bc
commit 44440ba
Showing 1 changed file with 8 additions and 12 deletions.
diff --git a/core/dbt/task/run.py b/core/dbt/task/run.py
@@ -110,18 +110,14 @@ def track_model_run(index, num_nodes, run_model_result):
     invocation_id = get_invocation_id()
     node = run_model_result.node
     has_group = True if hasattr(node, "group") and node.group else False
-    access = (
-        node.access.value
-        if (node.resource_type == NodeType.Model and node.access is not None)
-        else None
-    )
-    contract_enforced = (
-        node.contract.enforced
-        if (node.resource_type == NodeType.Model and node.contract)
-        else False
-    )
-    versioned = True if hasattr(node, "version") and node.version else False
-
+    if node.resource_type == NodeType.Model:
+        access = node.access.value if node.access is not None else None
+        contract_enforced = node.contract.enforced
+        versioned = True if node.version else False
+    else:
+        access = None
+        contract_enforced = False
+        versioned = False
     tracking.track_model_run(
         {
             "invocation_id": invocation_id,