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 = {