From 64e2ea6b69ee3b4f96069c9b332e40bf78055cef Mon Sep 17 00:00:00 2001
From: Peter Allen Webb <peter.webb@dbtlabs.com>
Date: Sun, 7 Jul 2024 15:06:00 -0400
Subject: [PATCH] Add group to record types.

---
 dbt/adapters/postgres/record.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/dbt/adapters/postgres/record.py b/dbt/adapters/postgres/record.py
index 09981b1..52248a7 100644
--- 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):