Python - Cambiar verbos a infinitivo de una string

import re

def replace_verbs(texto):

    texto2=texto.replace(",","").split()

    patron=["o", "as", "a", "amos", "áis", "an"]

    reemplazo="ar"

    for i in patron:

        return re.sub(f"{i}$", f"{reemplazo}$", texto)

#-*- coding: utf - 8 -*-

import re

cadena= "Trabajo, Judit canta, vosotros nadáis y ellos bailan "

def cambiar(texto):

    patron=["o", "as", "a", "amos", "áis", "an"]

    for x in patron:

        nueva = re.sub(rf"{x}\b","ar", texto)

        texto = nueva

    return nueva

print(cambiar(cadena))