Enhance: Improve the error message when the Metadata is empty. (#2257)

sdv-dev · Oct 8, 2024 · 241acf4 · 241acf4
1 parent 6ae4840
commit 241acf4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/sdv/metadata/metadata.py b/sdv/metadata/metadata.py
@@ -164,10 +164,7 @@ def _convert_to_single_table(self):
 
     def _handle_table_name(self, table_name):
         if len(self.tables) == 0:
-            raise ValueError(
-                'The metadata object is currently empty. To populate it, please use either '
-                "'detect_from_dataframe' or 'detect_from_dataframes'."
-            )
+            raise ValueError('Metadata does not contain any tables. No columns can be added.')
         if table_name is None:
             if len(self.tables) == 1:
                 table_name = next(iter(self.tables))

diff --git a/tests/unit/metadata/test_metadata.py b/tests/unit/metadata/test_metadata.py
@@ -714,9 +714,6 @@ def test__handle_table_name_with_empty_tables(self):
         instance = Metadata()
 
         # Run and Assert
-        error_msg = (
-            'The metadata object is currently empty. To populate it, please use either '
-            "'detect_from_dataframe' or 'detect_from_dataframes'."
-        )
+        error_msg = 'Metadata does not contain any tables. No columns can be added.'
         with pytest.raises(ValueError, match=error_msg):
             instance._handle_table_name(None)