remove post_serialize and fix test

dbt-labs · Dec 3, 2024 · f6c8715 · f6c8715
1 parent 878d5da
commit f6c8715
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/core/dbt/contracts/graph/nodes.py b/core/dbt/contracts/graph/nodes.py
@@ -1441,12 +1441,6 @@ def same_contents(self, old: Optional["Exposure"]) -> bool:
     def group(self):
         return None
 
-    def __post_serialize__(self, dct: Dict, context: Optional[Dict] = None):
-        dct = super().__post_serialize__(dct, context)
-        if "_event_status" in dct:
-            del dct["_event_status"]
-        return dct
-
 
 # ====================================
 # Metric node
@@ -1665,12 +1659,6 @@ def same_contents(self, old: Optional["SavedQuery"]) -> bool:
             and True
         )
 
-    def __post_serialize__(self, dct: Dict, context: Optional[Dict] = None):
-        dct = super().__post_serialize__(dct, context)
-        if "_event_status" in dct:
-            del dct["_event_status"]
-        return dct
-
 
 # ====================================
 # Patches

diff --git a/tests/unit/contracts/graph/test_nodes_parsed.py b/tests/unit/contracts/graph/test_nodes_parsed.py
@@ -2084,6 +2084,7 @@ def basic_parsed_exposure_dict():
             "enabled": True,
         },
         "unrendered_config": {},
+        "_event_status": {},
     }
 
 
@@ -2104,6 +2105,7 @@ def basic_parsed_exposure_object():
         tags=[],
         config=ExposureConfig(),
         unrendered_config={},
+        _event_status={},
     )
 
 
@@ -2139,6 +2141,7 @@ def complex_parsed_exposure_dict():
             "enabled": True,
         },
         "unrendered_config": {},
+        "_event_status": {},
     }
 
 
@@ -2162,6 +2165,7 @@ def complex_parsed_exposure_object():
         original_file_path="models/something.yml",
         config=ExposureConfig(),
         unrendered_config={},
+        _event_status={},
     )