def right_context(token_list, token, context_size, idx):
    right_window = []
    if idx >= len(token_list):
        #return ['_END_' for i in range(context_size)]
        return ['</s>' for i in range(context_size)]
    assert(token_list[idx] == token), "Token in token list: {}, index: {}, token provided in parameters: {}".format(token_list[idx], idx, token)
    for i in range(idx+1, idx+context_size+1):
        if i > len(token_list)-1:
            #right_window.append('_END_')
            right_window.append('</s>')
        else:
            right_window.append(token_list[i])
    return right_window
dataset/ETHPy150Open qe-team/marmot/marmot/features/word2vec_feature_extractor.py/right_context/original