def flatten()

in multiset_codec/rans.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def flatten(ans_state):
    """Flatten a vrans state ans_state into a 1d numpy array."""
    head, ans_state = np.ravel(ans_state[0]), ans_state[1]
    out = [np.uint32(head >> 32), np.uint32(head)]
    while ans_state:
        head, ans_state = ans_state
        out.append(head)
    return np.concatenate(out)