remove default labels column in favor of custom col name (#106)

Signed-off-by: Sarah Yurick <sarahyurick@gmail.com>
rapidsai · Dec 26, 2024 · 5968514 · 5968514
1 parent 25ecb0d
commit 5968514
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crossfit/op/label.py b/crossfit/op/label.py
@@ -57,7 +57,7 @@ def call(self, data: Union[cudf.Series, cudf.DataFrame]) -> Union[cudf.Series, c
         return output
 
     def meta(self):
-        labeled = {"labels": "string"}
+        labeled = {self.suffix: "string"}
 
         if self.cols and len(self.cols) > 1:
             labeled = {