diff --git a/morpheus/modules/payload_batcher.py b/morpheus/modules/payload_batcher.py index eb1b666dbf..29ab09fe5c 100644 --- a/morpheus/modules/payload_batcher.py +++ b/morpheus/modules/payload_batcher.py @@ -148,6 +148,8 @@ def _batch_dataframe_by_group(df: cudf.DataFrame) -> typing.List[cudf.DataFrame] df[period_column] = df[period_column].to_pandas().dt.to_period(period).astype('str') if len(group_by_columns) == 1: + # Avoid warning from cudf regardning an upcoming change of behavior when applying a groupby to a single + # element list. group_by_columns_ = group_by_columns[0] else: group_by_columns_ = group_by_columns