Error in Setting IDGenerator for Primary Key columns

[pvk-developer]

Environment Details

Please indicate the following details about the environment in which you found the bug:

• SDV version: 1.10
• Python version: any
• Operating System: any

Error Description

When trying to set an IDGenerator for a primary key we get an error saying that only AnonymizedFaker or RegexGenerator can be used however we do see this generator being used in some of our datasets when it comes to primary keys.

SynthesizerInputError: Invalid transformer 'IDGenerator' for a primary or alternate key 'user_id'. Please use 'AnonymizedFaker' or 'RegexGenerator' instead.

Steps to reproduce

import pandas as pd
from sdv.metadata import SingleTableMetadata
from sdv.single_table import GaussianCopulaSynthesizer
from rdt.transformers import IDGenerator

data = pd.DataFrame({
    'user_id': list(range(4)),
    'user_cat': ['a', 'b','c', 'd']
})

stm = SingleTableMetadata()
stm.detect_from_dataframe(data)
stm.update_column('user_id', sdtype='id')
stm.set_primary_key('user_id')

gc = GaussianCopulaSynthesizer(stm)
custom_id = IDGenerator(starting_value=5)
gc.auto_assign_transformers(data)

gc.update_transformers({'user_id': custom_id})

---------------------------------------------------------------------------
SynthesizerInputError                     Traceback (most recent call last)
Cell In[38], line 1
----> 1 gc.update_transformers({'user_id': custom_id})

File ~/Projects/sdv-dev/SDV/sdv/single_table/base.py:215, in BaseSynthesizer.update_transformers(self, column_name_to_transformer)
    213 self._validate_transformers(column_name_to_transformer)
    214 self._warn_for_update_transformers(column_name_to_transformer)
--> 215 self._data_processor.update_transformers(column_name_to_transformer)
    216 if self._fitted:
    217     msg = 'For this change to take effect, please refit the synthesizer using `fit`.'

File ~/Projects/sdv-dev/SDV/sdv/data_processing/data_processor.py:640, in DataProcessor.update_transformers(self, column_name_to_transformer)
    638 for column, transformer in column_name_to_transformer.items():
    639     if column in self._keys and not type(transformer) in (AnonymizedFaker, RegexGenerator):
--> 640         raise SynthesizerInputError(
    641             f"Invalid transformer '{transformer.__class__.__name__}' for a primary "
    642             f"or alternate key '{column}'. Please use 'AnonymizedFaker' or "
    643             "'RegexGenerator' instead."
    644         )
    646 with warnings.catch_warnings():
    647     warnings.filterwarnings('ignore', module='rdt.hyper_transformer')

SynthesizerInputError: Invalid transformer 'IDGenerator' for a primary or alternate key 'user_id'. Please use 'AnonymizedFaker' or 'RegexGenerator' instead.