remove train_test_split backport. It wasn't used consistently and sklearn refactoring breaks the current backport.
Remove error in __repr__ when n_states is none.