Python Function Define Input Data Type