Fix order in which transformers are returned (#1239)

sdv-dev · Mar 23, 2023 · ffca169 · ffca169
1 parent d4bba60
commit ffca169
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/sdv/single_table/base.py b/sdv/single_table/base.py
@@ -352,6 +352,11 @@ def get_transformers(self):
                 "Use 'auto_assign_transformers' or 'fit' to create them."
             )
 
+        field_transformers = {
+            column_name: field_transformers.get(column_name)
+            for column_name in self.metadata._columns
+        }
+
         return field_transformers
 
     def get_info(self):

diff --git a/tests/unit/single_table/test_base.py b/tests/unit/single_table/test_base.py
@@ -146,14 +146,18 @@ def test_get_transformers(self):
             'name': 'FrequencyEncoder',
             'salary': 'FloatFormatter'
         }
+        instance.metadata._columns = {
+            'salary': {'sdtype': 'numerical'},
+            'name': {'sdtype': 'categorical'}
+        }
 
         # Run
         result = BaseSingleTableSynthesizer.get_transformers(instance)
 
         # Assert
         assert result == {
-            'name': 'FrequencyEncoder',
-            'salary': 'FloatFormatter'
+            'salary': 'FloatFormatter',
+            'name': 'FrequencyEncoder'
         }
 
     def test_get_transformers_raises_an_error(self):