in src/mozanalysis/experiment.py [0:0]
def get(self, bq_context: BigQueryContext, analysis_window) -> DataFrame:
"""Get the DataFrame for a specific analysis window.
N.B. this makes a BigQuery query each time it is run; caching
results is your responsibility.
Args:
bq_context (BigQueryContext)
analysis_window (AnalysisWindow or int): The analysis
window, or its start day as an int.
"""
if isinstance(analysis_window, int):
try:
analysis_window = next(
aw for aw in self.analysis_windows if aw.start == analysis_window
)
except StopIteration as err:
raise KeyError(
f"AnalysisWindow not found with start of {analysis_window}"
) from err
return bq_context.run_query(
self._build_analysis_window_subset_query(analysis_window)
).to_dataframe()