C3 W1 Assignment 2 Excercise 5: Assertion error

Whenever I test the model, the follwing error accures:
AssertionError: Fail in example 1. Expected -0.699999988079071 but got -0.0
Could it be due to an issue with using tf.one_hot?

There is comment in exercise 5:

  • To convert the target into the same dimension as the predictions tensor use tf.one_hot with target and preds.shape[-1].

So one_hot can be used!