Adjust sort

hi-primus · Oct 20, 2021 · e18711c · e18711c
1 parent 1768f4f
commit e18711c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/optimus/engines/base/rows.py b/optimus/engines/base/rows.py
@@ -189,10 +189,11 @@ def _set_order(o):
         dfd = df.functions.sort_df(df.data, sort_cols, order)
         meta = Meta.action(self.root.meta, Actions.SORT_ROW.value, cols)
 
+        if cast:
+            dfd = dfd.drop(sort_cols, axis=1)
+
         df = self.root.new(dfd, meta=meta)
 
-        if cast:
-            df = df.cols.drop(sort_cols)
 
         return df