Python Function Define Input Variable In Flow