error in Listing 2.9 (`one_training_step()`, § Running one training step, ch. 2)

[rgiulietti]

The code computes average_loss but never uses it. On the other hand, the invocation of get_gradients_of_loss_wrt_weights() passes loss instead of average_loss as the 1st argument.