Address comments

sdv-dev · Sep 12, 2022 · eae3bcb · eae3bcb
1 parent 215036d
commit eae3bcb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/rdt/hyper_transformer.py b/rdt/hyper_transformer.py
@@ -809,7 +809,6 @@ def _reverse_transform(self, data, prevent_subset):
             is_subset = contained and len(data.columns) < len(self._output_columns)
             if is_subset:
                 raise Error(
-                    'There are unexpected columns in the data you are trying to transform. '
                     'You must provide a transformed dataset with all the columns from the '
                     'original data.'
                 )

diff --git a/tests/unit/test_hyper_transformer.py b/tests/unit/test_hyper_transformer.py
@@ -1718,8 +1718,7 @@ def test_reverse_transform_with_subset(self):
 
         # Run / Assert
         expected_msg = (
-            'There are unexpected columns in the data you are trying to transform. You must '
-            'provide a transformed dataset with all the columns from the original data.'
+            'You must provide a transformed dataset with all the columns from the original data.'
         )
         with pytest.raises(Error, match=expected_msg):
             ht.reverse_transform(data)