Add group to record types.

dbt-labs · Jul 7, 2024 · 64e2ea6 · 64e2ea6
1 parent 492b412
commit 64e2ea6
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/dbt/adapters/postgres/record.py b/dbt/adapters/postgres/record.py
@@ -6,7 +6,6 @@
 
 
 class PostgresRecordReplayHandle(RecordReplayHandle):
-    # PAW: Wrap
     def cursor(self):
         cursor = None if self.native_handle is None else self.native_handle.cursor()
         return PostgresRecordReplayCursor(cursor, self.connection)
@@ -22,12 +21,11 @@ class CursorGetStatusMessageResult:
     msg: Optional[str]
 
 
+@Recorder.register_record_type
 class CursorGetStatusMessageRecord(Record):
     params_cls = CursorGetStatusMessageParams
     result_cls = CursorGetStatusMessageResult
-
-
-Recorder.register_record_type(CursorGetStatusMessageRecord)
+    group = "Database"
 
 
 class PostgresRecordReplayCursor(RecordReplayCursor):