Replace input function with another function

replace_decorated_function.py

def decorator(func):
  return add
  
def add(a, b):
  return a+b

def multiply(a, b):
  return a*b

multiply = decorator(multiply)

print(multiply(8,8)) #16