in sparse_autoencoder/explanations.py [0:0]
def a_star_c_fn(tokens: list[list[float]]) -> list[list[float]]:
return [
[1.0 if (tok == 2 and 0 in toks[:i]) else 0.0 for i, tok in enumerate(toks)]
for toks in tokens
]