# This notebook requires gbstats version 0.7.0 or later
try:
    import gbstats.utils
    gbstats.utils.check_gbstats_compatibility('0.7.0')
except ModuleNotFoundError:
    raise ValueError('Upgrade gbstats to 0.7.0 or later from PyPI using `pip install gbstats`')

from gbstats.gbstats import (
    filter_query_rows,
    process_analysis,
    AnalysisSettingsForStatsEngine,
    MetricSettingsForStatsEngine,
)
import pandas as pd

# Mapping of variation id to index
var_id_map = {'0': 0, '1': 1}

# Analysis settings
analysis = AnalysisSettingsForStatsEngine(var_names=['Control', 'Variation 1'], var_ids=['0', '1'], weights=[0.5, 0.5], baseline_index=0, dimension='', stats_engine='frequentist', sequential_testing_enabled=False, sequential_tuning_parameter=5000.0, difference_type='relative', phase_length_days=16.083333333333332, alpha=0.05, max_dimensions=20)

# Columns to show in the result summary
summary_cols = ['dimension', 'baseline_name', 'baseline_users', 'baseline_cr', 'v1_name', 'v1_users', 'v1_cr', 'v1_expected', 'v1_ci', 'v1_p_value']