fix previous commit

NVIDIA-Merlin · Apr 11, 2024 · f4c9de3 · f4c9de3
1 parent f091625
commit f4c9de3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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