Python Type Annotation Function As Parameter