Added reset function

eapframework · Nov 6, 2019 · 7a3c113 · 7a3c113
1 parent 05a105b
commit 7a3c113
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/optimus/dataframe/extension.py b/optimus/dataframe/extension.py
@@ -436,6 +436,12 @@ def debug(self):
     """
     print(self.rdd.toDebugString().decode("ascii"))
 
+@add_method(DataFrame)
+def reset(self):
+
+    df = self.set_meta("transformations.actions", {})
+    Profiler.instance.output_columns = {}
+    return df
 
 @add_method(DataFrame)
 def send(self, name=None, infer=True, mismatch=None, stats=True):