def _for_interval_sample_size_calculation()

in src/mozanalysis/frequentist_stats/sample_size.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

    def _for_interval_sample_size_calculation(i):
        df_interval = df.loc[df["enrollment_date"] < interval_end_dates[i]]
        res = z_or_t_ind_sample_size_calc(
            df=df_interval, metrics_list=metric_list, test="t", **sizing_kwargs
        )
        final_res = {}
        for key in res:
            final_res[key] = {
                "enrollment_end_date": interval_end_dates[i],
                **res[key],
            }

        return final_res