change 1 04-05-2025

Author: R-Palazzo

sdv/multi_table/base.py

@@ -164,9 +164,6 @@ def add_cag(self, patterns):
             patterns (list):
                 A list of CAG patterns to apply to the synthesizer.
         """
-        if not hasattr(self, '_original_metadata'):
-            self._original_metadata = self.metadata
-
         metadata = self.metadata
         for pattern in patterns:
             metadata = pattern.get_updated_metadata(metadata)
@@ -330,10 +327,7 @@ def _validate_cags(self, data):
             data (pandas.DataFrame):
                 The data to validate.
         """
-        metadata = self.metadata
-        if hasattr(self, '_original_metadata'):
-            metadata = self._original_metadata
-
+        metadata = getattr(self, '_original_metadata', self.metadata)
         if hasattr(self, 'patterns'):
             for pattern in self.patterns:
                 pattern.validate(data=data, metadata=metadata)

sdv/single_table/base.py

@@ -733,7 +733,7 @@ def _validate_cags(self, data):
             data (pandas.DataFrame):
                 The data to validate.
         """
-        metadata = self._original_metadata
+        metadata = getattr(self, '_original_metadata', self.metadata)
         if hasattr(self, '_chained_patterns'):
             for pattern in self._chained_patterns:
                 pattern.validate(data=data, metadata=metadata)

tests/unit/multi_table/test_base.py

@@ -1478,6 +1478,7 @@ def test_add_cag(self):
         delattr(instance, 'patterns')
         original_metadata = get_multi_table_metadata()
         instance.metadata = original_metadata
+        instance._original_metadata = original_metadata
         pattern1 = Mock()
         pattern2 = Mock()
         patterns = [pattern1, pattern2]