Creating a custom callback in Keras to log gradients to Tensorboard

Question

I am trying to write a Keras callback that can log gradients to Tensorboard. However I run into the following error.

Traceback (most recent call last):
  File