address feedback

Author: gsheni

sdv/multi_table/base.py

@@ -476,11 +476,10 @@ def _assign_table_transformers(self, synthesizer, table_name, table_data):
         column_name_to_transformers = {}
 
         primary_key = self.metadata.tables[table_name].primary_key
-        foreign_keys = self.metadata._get_all_foreign_keys(table_name)
-        for foreign_key in foreign_keys:
-            if foreign_key != primary_key:
-                column_name_to_transformers[foreign_key] = None
-
+        foreign_key_columns = self.metadata._get_all_foreign_keys(table_name)
+        column_name_to_transformers = {
+            column_name: None for column_name in foreign_key_columns if column_name != primary_key
+        }
         synthesizer.update_transformers(column_name_to_transformers)
 
     def auto_assign_transformers(self, data):

sdv/multi_table/hma.py

@@ -788,10 +788,14 @@ def _add_foreign_key_columns(self, child_table, parent_table, child_name, parent
         foreign key values (from the parent table's PK).
 
         Args:
-            child_table (pd.DataFrame): The child table which may or may not contain the FK columns.
-            parent_table (pd.DataFrame): The parent table.
-            child_name (str): The name of the child table in the metadata.
-            parent_name (str): The name of the parent table in the metadata.
+            child_table (pd.DataFrame):
+                The child table which may or may not contain the FK columns.
+            parent_table (pd.DataFrame):
+                The parent table which. contains the primary key ID values.
+            child_name (str):
+                The name of the child table in the metadata.
+            parent_name (str):
+                The name of the parent table in the metadata.
 
         Returns:
             None: The child_table is modified in-place.