Partial Function Python Code