import fused

@fused.udf
def my_udf(variables):
    import os
    import math

    array_numbers = variables['array_numbers']
    first_number = variables['first_number']
    second_number = variables['second_number']
    path = variables['path']

    suma_from_array = 0
    for num in array_numbers:
        suma_from_array = suma_from_array + num

    suma_from_numbers = first_number + second_number
    square_root_from_numbers = math.sqrt(suma_from_numbers)

    new_path = path.replace('example','my_function')

    object_to_return = {"suma_from_array":suma_from_array,
                        "square_root_from_numbers": square_root_from_numbers,
                        "new_path": new_path}
    return object_to_return

# Save locally
my_udf.to_directory('/home/data3/master_folder/fused/repositorios/operatios_with_numbers')