使用Apply函数时如何解决类型错误

我试图在表中的列上使用apply函数。这是示例数据和代码。

df = pd.DataFrame({'a':[0,1,2], 'text':["I love a dog", "He is smart", "I hate this"]})

def edit(input_text):
    output = input_text[:4]
    return output


df['text'].apply(lambda x: edit(x))

我以为它将在该列的每一行上应用编辑功能,但这给了我类型错误。

请帮帮我。

谢谢。