custom python script as azure end point?

what would be the code for scoring.py on azure end point, for the following python function
def predict(a):
return a*a