Python Assign Function