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