Python: Typing a lambda function for use with sorted