diff --git a/optimus/dataframe/extension.py b/optimus/dataframe/extension.py index 0a8d8ddf..8b077d3d 100644 --- 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):