Update hma.py

sdv-dev · Jun 18, 2024 · 53743f0 · 53743f0
1 parent c55dd06
commit 53743f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdv/multi_table/hma.py b/sdv/multi_table/hma.py
@@ -273,7 +273,7 @@ def preprocess(self, data):
 
     def _set_extended_columns_distributions(self, synthesizer, table_name, valid_columns):
         numerical_distributions = {}
-        for extended_column in self._parent_extended_columns.get(table_name, []):
+        for extended_column in self._parent_extended_columns[table_name]:
             if extended_column in valid_columns:
                 numerical_distributions[extended_column] = DEFAULT_EXTENDED_COLUMNS_DISTRIBUTION
         synthesizer._set_numerical_distributions(numerical_distributions)