Skip to content

Commit

Permalink
Adding a note to custom_df in aggregate
Browse files Browse the repository at this point in the history
  • Loading branch information
NivekT committed Jan 3, 2024
1 parent 9cf3319 commit c4b5c7d
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions prompttools/harness/harness.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,6 +104,9 @@ def aggregate(
groupby_column (str):
aggregate_columns (Union[str, list[str]]):
method (str): aggregation method (e.g., 'mean', 'sum', 'count', 'min', 'max', 'median', 'std', etc.)
custom_df (Optional[pd.DataFrame]): By default, this method uses `self.full_df`, however, a specific
subclass implementation can chooses to override this (mainly to make modification before or after
group by)
"""
if method not in ["mean", "sum", "count", "min", "max", "median", "std"]:
raise ValueError(f"Unsupported aggregation method: {method}")
Expand Down

0 comments on commit c4b5c7d

Please sign in to comment.