def get_ate(data_ps):
    """Get ATE without bias correction
    Args:
        data_ps (pandas.DataFrame): dataframe with propensity score
    Returns:
        float: average treatment effect
    """
    result = data_ps.groupby("nudge")["outcome"].mean()
    ate = result[1] - result[0]
    # print("Calculate Average Treatment Effect:")
    # print(f"Control: {round(result[0], 3)}")
    # print(f"Treatment: {round(result[1], 3)}")
    # print(f"ATE: {round(ate, 3)}")

    return ate