def __init__(self, grouped_data, agg_cols):
    self.grouped_data = grouped_data
    self.agg_cols = agg_cols
def __init__(self, grouped_data, agg_cols):
    self.grouped_data = grouped_data
    self.agg_cols = agg_cols