From b5df1318c6dcce4a8730b788961328e52b98e21b Mon Sep 17 00:00:00 2001
From: Sarah Yurick <sarahyurick@gmail.com>
Date: Tue, 24 Dec 2024 10:23:38 -0800
Subject: [PATCH] remove default labels column in favor of custom col name
Signed-off-by: Sarah Yurick <sarahyurick@gmail.com>
---
crossfit/op/label.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/crossfit/op/label.py b/crossfit/op/label.py
index b549dc7..c8ed06a 100644
--- 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 = {