data_loaders.py

@config.when(mode="online")
@extract_columns(
    "budget",
    "age",
    "gender",
)
def survey_results__online(client_id: int) -> pd.DataFrame:
    """Map operation to explode survey results to all fields
    Data comes in JSON, we've grouped it into a series.
    """
    return utils.query_survey_results(client_id=client_id)
  
@config.when(mode="online")
def last_logged_in__online(client_id: int) -> pd.Series:
    return utils.query_login_data(client_id=client_id)["last_logged_in"]