Python os.walk carpetas en bucle en un patrón para cambiar el nombre

Hola, estoy tratando de aprender el módulo os.walk.

Gracias de antemano

Objetivo: cambiar el nombre de todas las carpetas en 'carpeta de estudiante' Para leer 1, 2, 3, 4

Problema: mi código solo cambia el nombre de la primera carpeta a 1 con éxito, los demás siguen siendo los mismos.

for root, sub, files in os.walk(my_dir):
 for x in sub:
 count = 0
 new_name = count + 1
 os.rename(f'{root}/{x}', f'{root}/{new_name}')

Solución del problema

Parece que nunca aumentas el conteo. También podría eliminar la new_namevariable por completo e ir por algo como esto:

for root, sub, files in os.walk(my_dir):
 count = 0 
 for x in sub:
 count += 1
 os.rename(f'{root}/{x}', f'{root}/{count}')