Update crossfit/op/base.py

Co-authored-by: Vibhu Jawa <[email protected]>
rapidsai · Sep 25, 2024 · 2dd6e68 · 2dd6e68
1 parent 4c9c2b3
commit 2dd6e68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crossfit/op/base.py b/crossfit/op/base.py
@@ -70,7 +70,7 @@ def add_keep_cols(self, data, output):
                 output[col] = data[col]
 
         columns = list(output.columns)
-        # we use dict.fromkeys to remove duplicates and preserve order
+        # we use dict.fromkeys to remove duplicates and preserve order (to match _build_dask_meta behavior) 
         output = output[list(dict.fromkeys(self.keep_cols + columns))]
 
         return output