def tags_from_contexts(token_contexts):
    return {token: np.array([context['tag'] for context in contexts]) for token, contexts in token_contexts.items()}
create a dict mapping tokens to their tags
dataset/ETHPy150Open qe-team/marmot/marmot/experiment/experiment_utils.py/tags_from_contexts