be more specific:

        num_states, num_vecs = ...