I'm getting `sort` got unexpected keyword argument `key`. I don't think numpy rows can be sorted like that. Looks like an awesome library otherwise!