Fixed values_to_cols() metadata

eapframework · Nov 8, 2019 · 00b6ba2 · 00b6ba2
1 parent b2fe7d3
commit 00b6ba2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/optimus/dataframe/columns.py b/optimus/dataframe/columns.py
@@ -2113,10 +2113,15 @@ def join_all(_dfs):
 
             # names = before.cols.names(keys, invert=True)
             # print(names)
+            pivotDF = pivotDF.preserve_meta(self)
             df = pivotDF.toDF(*names).cols.fill_na(new_names, 0)
-            # df.table()
+            df = df.preserve_meta(self, Actions.VALUES_TO_COLS.value, new_names)
+
             combined.append(df)
-        return join_all(combined)
+
+        df = join_all(combined)
+
+        return df
 
     @add_attr(cols)
     def string_to_index(input_cols=None, output_cols=None, columns=None):

diff --git a/optimus/helpers/constants.py b/optimus/helpers/constants.py
@@ -103,6 +103,7 @@ class Actions(Enum):
     NEST = "nest"
     UNNEST = "unnest"
     DROP_ROW = "drop_row"
+    VALUES_TO_COLS = "values_to_cols"
 
     @staticmethod
     def list():