Fix table as well

Author: fealho

sdv/multi_table/hma.py

@@ -271,7 +271,7 @@ def _print_estimate_warning(self):
             entry.append(table)
             entry.append(sum(metadata_columns[table]))
             total_est_cols += est_cols
-            entry.append(est_cols)
+            entry.append(min(est_cols, PERFORMANCE_ALERT_DISPLAY_CAP))
             print_table.append(entry)
 
         if total_est_cols > MAX_NUMBER_OF_COLUMNS:

tests/unit/multi_table/test_hma.py

@@ -148,7 +148,7 @@ def test__print_estimate_warning_many_cols(self, get_distributions_mock, estimat
             'PerformanceAlert: Using the HMASynthesizer on this metadata schema is not recommended.'
             ' To model this data, HMA will generate a large number of columns. (1000000+ columns)\n'
             '\n\nTable Name  # Columns in Metadata  Est # Columns\n'
-            '   nesreca                      1        1000010\n\n'
+            '   nesreca                      1        1000000\n\n'
             "We recommend simplifying your metadata schema using 'sdv.utils.poc.simplify_schema'."
             '\nIf this is not possible, please visit datacebo.com and reach out to us for '
             'enterprise solutions.\n\n'