Fix lint

sdv-dev · Feb 16, 2023 · 1f01f0d · 1f01f0d
1 parent 195617f
commit 1f01f0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdv/multi_table/hma.py b/sdv/multi_table/hma.py
@@ -248,7 +248,7 @@ def _finalize(self, sampled_data):
                                 foreign_key,
                                 sampled_data
                             )
-                            table_rows[foreign_key] = parent_ids.values
+                            table_rows[foreign_key] = parent_ids.to_numpy()
 
             synthesizer = self._table_synthesizers.get(table_name)
             dtypes = synthesizer._data_processor._dtypes