Ayuda a Bobby a corregir su código

Bobby está intentando resolver un desafío en Edabit y necesita depurar su código. Debe crear una función que reciba como argumento el tamaño de una matriz cuadrada. La función debe devolver la matriz con las diagonales establecidas en 1 y todos los demás elementos en 0. Su código está en la pestaña Code, pero contiene errores en dos líneas. ¿Puedes ayudarlo?

Ejemplos

help_bobby(1) ➞ [[1]]

help_bobby(3) ➞ [
  [1, 0, 1],
  [0, 1, 0],
  [1, 0, 1]
]

help_bobby(4) ➞ [
  [1, 0, 0, 1],
  [0, 1, 1, 0],
  [0, 1, 1, 0],
  [1, 0, 0, 1]
]

Notas

N/A