remove default labels column in favor of custom col name

Signed-off-by: Sarah Yurick <[email protected]>
rapidsai · Dec 24, 2024 · b5df131 · b5df131
1 parent 25ecb0d
commit b5df131
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 = {