docstring

sdv-dev · Sep 8, 2023 · 9153334 · 9153334
1 parent acfd2c8
commit 9153334
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rdt/transformers/base.py b/rdt/transformers/base.py
@@ -488,8 +488,8 @@ class BaseMultiColumnTransformer(BaseTransformer):
     in order to create a new multi column transformer.
 
     Attributes:
-        columns_to_sdtype (tuple):
-            Order of the columns to be used for the transformer.
+        columns_to_sdtype (dict):
+            Dictionary mapping each column to its sdtype.
         prefixes (dict):
             Dictionary mapping each output column to its prefix.
     """
@@ -572,8 +572,8 @@ def fit(self, data, columns_to_sdtype):
         Args:
             data (pandas.DataFrame):
                 The entire table.
-            columns_to_sdtype (tuple):
-                Order of the columns to be used for the transformer.
+            columns_to_sdtype (dict):
+                Dictionary mapping each column to its sdtype.
         """
         self._validate_columns_to_sdtype(data, columns_to_sdtype)
         self.columns_to_sdtype = columns_to_sdtype
@@ -589,8 +589,8 @@ def fit_transform(self, data, columns_to_sdtype):
         Args:
             data (pandas.DataFrame):
                 The entire table.
-            columns_to_sdtype (tuple):
-                Order of the columns to be used for the transformer.
+            columns_to_sdtype (dict):
+                Dictionary mapping each column to its sdtype.
 
         Returns:
             pd.DataFrame: