From f4c9de32e9720fe7043fcaf5c17bb21b111dff80 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Thu, 11 Apr 2024 09:10:21 -0700 Subject: [PATCH] fix previous commit --- nvtabular/ops/groupby.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nvtabular/ops/groupby.py b/nvtabular/ops/groupby.py index 1b6084b15e..93748e4f19 100644 --- a/nvtabular/ops/groupby.py +++ b/nvtabular/ops/groupby.py @@ -109,9 +109,12 @@ def __init__( self.list_aggs[col] = list(_list_aggs) self.name_sep = name_sep - self.supported_formats = DataFormats.PANDAS_DATAFRAME | DataFormats.CUDF_DATAFRAME super().__init__() + @property + def supported_formats(self): + return DataFormats.PANDAS_DATAFRAME | DataFormats.CUDF_DATAFRAME + @annotate("Groupby_op", color="darkgreen", domain="nvt_python") def transform(self, col_selector: ColumnSelector, df: DataFrameType) -> DataFrameType: # Sort if necessary