如何用Python编写函数

用Python编写函数，判断输入的两个数字的最大值，该如何编写代码呢？

def func(a, b):
    if a>b:
        print(a)
    else:
        print(b)

def find_max(a,b):
    return max(a, b)


if __name__ == '__main__':
    max_num = find_max(a=5, b=3)
    print(max_num)