[Enterprise Usage] Unable to assign generic PII transformers

rdt/transformers/pii/anonymizer.py

@@ -124,6 +124,23 @@ def __init__(self, provider_name=None, function_name=None, function_kwargs=None,
         self.missing_value_generation = missing_value_generation
         self._nan_frequency = 0.0
 
+    @classmethod
+    def get_supported_sdtypes(cls):
+        """Return the supported sdtypes by the transformer.
+
+        Returns:
+            list:
+                Accepted input sdtypes of the transformer.
+        """
+        unsupported_sdtypes = {'numerical', 'datetime', 'categorical', 'boolean', 'text', None}
+        all_sdtypes = {cls.INPUT_SDTYPE}
+        for transformer in BaseTransformer.get_subclasses():
+            if not issubclass(transformer, cls):
+                all_sdtypes.update(transformer.get_supported_sdtypes())
+
+        supported_sdtypes = all_sdtypes - unsupported_sdtypes
+        return list(supported_sdtypes)
+
     def reset_randomization(self):
         """Create a new ``Faker`` instance."""
         super().reset_randomization()

tests/integration/transformers/pii/test_anonymizer.py

@@ -27,6 +27,15 @@ def test_anonymizedfaker():
     assert len(reverse_transform['username']) == 5
 
 
+def test_anonymizedfaker_get_supported_sdtypes():
+    """Test that the correct supported sdtypes are returned."""
+    # Run
+    supported_sdtypes = AnonymizedFaker.get_supported_sdtypes()
+
+    # Assert
+    assert supported_sdtypes == ['pii']
+
+
 def test_anonymizedfaker_custom_provider():
     """End to end test with a custom provider and function for the ``AnonymizedFaker``."""
     data = pd.DataFrame({

tests/unit/transformers/pii/test_anonymizer.py

@@ -298,6 +298,42 @@ def test___init__no_function_name(self):
         with pytest.raises(TransformerInputError, match=expected_message):
             AnonymizedFaker(provider_name='credit_card', locales=['en_US', 'fr_FR'])
 
+    @patch('rdt.transformers.pii.anonymizer.issubclass')
+    @patch('rdt.transformers.pii.anonymizer.BaseTransformer')
+    def test_get_supported_sdtypes(self, base_mock, issubclass_mock):
+        """Test that the method returns all sdtypes except the basic ones."""
+        # Setup
+        issubclass_mock.return_value = False
+        numerical_mock = Mock()
+        numerical_mock.get_supported_sdtypes.return_value = ['numerical']
+        categorical_mock = Mock()
+        categorical_mock.get_supported_sdtypes.return_value = ['categorical']
+        datetime_mock = Mock()
+        datetime_mock.get_supported_sdtypes.return_value = ['datetime']
+        boolean_mock = Mock()
+        boolean_mock.get_supported_sdtypes.return_value = ['boolean', 'categorical']
+        text_mock = Mock()
+        text_mock.get_supported_sdtypes.return_value = ['text']
+        phone_mock = Mock()
+        phone_mock.get_supported_sdtypes.return_value = ['phone_number']
+        pii_mock = Mock()
+        pii_mock.get_supported_sdtypes.return_value = ['pii']
+        base_mock.get_subclasses.return_value = [
+            numerical_mock,
+            categorical_mock,
+            datetime_mock,
+            boolean_mock,
+            text_mock,
+            phone_mock,
+            pii_mock
+        ]
+
+        # Run
+        supported_sdtypes = AnonymizedFaker.get_supported_sdtypes()
+
+        # Assert
+        assert sorted(supported_sdtypes) == sorted(['phone_number', 'pii'])
+
     @patch('rdt.transformers.pii.anonymizer.BaseTransformer.reset_randomization')
     @patch('rdt.transformers.pii.anonymizer.faker')
     def test_reset_randomization(self, mock_faker, mock_base_reset):