Skip to content

Commit 101a8bb

Browse files
committed
variable word choice
1 parent 81d9342 commit 101a8bb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sdv/multi_table/hma.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,9 @@ def _estimate_num_columns(cls, metadata, distributions=None):
172172
metadata, table_name, columns_per_table, visited, distributions
173173
)
174174

175-
return {key: sum(value) for key, value in columns_per_table.items()}
175+
return {
176+
table_name: sum(columns_list) for table_name, columns_list in columns_per_table.items()
177+
}
176178

177179
def __init__(self, metadata, locales=['en_US'], verbose=True):
178180
BaseMultiTableSynthesizer.__init__(self, metadata, locales=locales)

0 commit comments

Comments
 (0)