Skip to content

Commit 95de140

Browse files
committed
handle missing table name
1 parent 8405900 commit 95de140

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

sdv/multi_table/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -404,6 +404,7 @@ def set_table_parameters(self, table_name, table_parameters):
404404
self._table_synthesizers[table_name] = self._synthesizer(
405405
metadata=table_metadata, **table_parameters
406406
)
407+
self._table_synthesizers[table_name]._data_processor.table_name = table_name
407408
self._table_parameters[table_name].update(deepcopy(table_parameters))
408409

409410
def _validate_all_tables(self, data):

sdv/multi_table/hma.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -185,14 +185,14 @@ def __init__(self, metadata, locales=['en_US'], verbose=True):
185185
self._default_parameters = {}
186186
self._parent_extended_columns = defaultdict(list)
187187
self.verbose = verbose
188+
BaseHierarchicalSampler.__init__(
189+
self, self.metadata, self._table_synthesizers, self._table_sizes
190+
)
188191
child_tables = set()
189192
for relationship in metadata.relationships:
190193
child_tables.add(relationship['child_table_name'])
191194
for child_table_name in child_tables:
192195
self.set_table_parameters(child_table_name, {'default_distribution': 'norm'})
193-
BaseHierarchicalSampler.__init__(
194-
self, self.metadata, self._table_synthesizers, self._table_sizes
195-
)
196196
self._print_estimate_warning()
197197

198198
def set_table_parameters(self, table_name, table_parameters):

0 commit comments

Comments
 (0)