Algoritmia - ayudenme con este problema

# Función para calcular el promedio de las tres notas más altas

def promedio_tres_mas_altas(notas):

    # Ordenar las notas de mayor a menor

    notas_ordenadas = sorted(notas, reverse=True)

    # Tomar las tres notas más altas

    tres_mas_altas = notas_ordenadas[:3]

    # Calcular el promedio de las tres notas más altas

    promedio = sum(tres_mas_altas) / 3

    return promedio

# Ingresar las cinco notas

nota1 = float(input("Ingrese la primera nota: "))

nota2 = float(input("Ingrese la segunda nota: "))

nota3 = float(input("Ingrese la tercera nota: "))

nota4 = float(input("Ingrese la cuarta nota: "))

nota5 = float(input("Ingrese la quinta nota: "))

# Crear una lista con las cinco notas

notas = [nota1, nota2, nota3, nota4, nota5]

# Calcular el promedio de las tres notas más altas

promedio_final = promedio_tres_mas_altas(notas)

# Imprimir el resultado

print("El promedio de las tres notas más altas es:", promedio_final)