Выскакивает ошибка:TypeError: '<' not supported between instances of 'list' and 'int'

Нужно написать функцию которая принимает в себя элементы списка в качестве аргумента и возвращает его с элементами число которых не превышает 20 Если делать без функции тобиш так:

j=[2,4,16,32,64]
j2 = [i for i in j if i < 21]
print(j2)

То это все работает Но если через функцию:

def function(*array):
  finaly_array = [i for i in array if i < 21]  
  return finaly_array 

function([2,4,6,8,10,20,22,40,142])

То выбивает ошибку: TypeError: '<' not supported between instances of 'list' and 'int'