Revert "[fix] Use rendered query comment for job labels (#955)"

This reverts commit d5b4114.
dbt-labs · Oct 6, 2023 · 8c0a7f4 · 8c0a7f4
1 parent d5b4114
commit 8c0a7f4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 61 deletions.
diff --git a/.changes/unreleased/Fixes-20231005-235950.yaml b/.changes/unreleased/Fixes-20231005-235950.yaml
diff --git a/dbt/adapters/bigquery/connections.py b/dbt/adapters/bigquery/connections.py
@@ -451,10 +451,9 @@ def raw_execute(
             hasattr(self.profile, "query_comment")
             and self.profile.query_comment
             and self.profile.query_comment.job_label
-            and self.query_header
-            and (query_comment := self.query_header.comment.query_comment)
         ):
-            labels = self._labels_from_query_comment(query_comment)
+            query_comment = self.profile.query_comment
+            labels = self._labels_from_query_comment(query_comment.comment)
         else:
             labels = {}
 

diff --git a/tests/functional/adapter/query_comment_test/test_job_label.py b/tests/functional/adapter/query_comment_test/test_job_label.py