Shkruani një kod Python për të marrë maksimumin e dy numrave.
def max(a, b):
""" Funksioni për të marrë maksimumin e dy numrave

Ky funksion do të marrë dy numra si argumente dhe
ktheni maksimumin e të dyve.

:param a: Numri i parë.
:param b: Numri i dytë.
:return: Maksimumi i dy numrave.
"""
    nëse a > b:
        ktheje a
    tjetër:
        kthimi b