I think there is an error in this course

I wrote a code that works, but I can not get it to compile the way it's asking me to. Is this an error, or am I doing it wrong?

In the editor define a function named map with two generic type parameters, T and U. The function takes two arguments - array of type [T], and transformation of type (T) -> U. The function returns an array of type U. The goal of the function is to apply the transformation operation on the array passed in as argument to return a new array containing the values transformed. For example given the array [1,2,3,4,5] as first argument, and a square function as the transformation argument, the result should be [1, 4, 9, 16, 25].

func map<T, U>(array: T, transformation: (T) -> U ) -> [U] {
    return [transformation(array)]
    }


func squareArray(_ value: [Int]) -> [Int] {

    var newArray: [Int] = []

    for n in value {

        newArray.append(n * n)
    }

   return newArray
}