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