align_state import from nltools fails

[jcheong0428]

My fnl_tools build is failing because this line is trying to import align_states from nltools which doesn't exist.

I only see align in nltools.stats at the moment: https://github.com/cosanlab/nltools/blob/91822a45778415ee2cdded7134e60bcde2bb7814/nltools/stats.py#L32

Is align_states a new function planned to be added to nltools?