comment

sdv-dev · Aug 17, 2023 · 6f8a243 · 6f8a243
1 parent c136cf6
commit 6f8a243
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdv/data_processing/data_processor.py b/sdv/data_processing/data_processor.py
@@ -466,7 +466,7 @@ def _create_config(self, data, columns_created_by_constraints):
                         is_numeric
                     )
                     sdtypes[column] = 'text'
-                elif column == self.metadata.primary_key or column in self.metadata.alternate_keys:
+                elif column in self._keys:
                     prefix = None
                     if not is_numeric:
                         prefix = 'sdv-id-'

diff --git a/tests/unit/data_processing/test_data_processor.py b/tests/unit/data_processing/test_data_processor.py
@@ -1069,7 +1069,7 @@ def test__create_config(self):
         dp.metadata.primary_key = 'id'
         dp.metadata.alternate_keys = ['id_no_regex', 'id_numeric']
         dp._primary_key = 'id'
-        dp._keys = ['id']
+        dp._keys = ['id', 'id_no_regex', 'id_numeric']
         dp.metadata.columns = {
             'int': {'sdtype': 'numerical'},
             'float': {'sdtype': 'numerical'},