C5_W4_A1_Transformer_Subclass_v1_Excercise 5

In Course5, Week 4, Assignment 1, on #UNQ_C5 cell, I get the following error and cannot figure it out. Any help is appreciated.

In step 2, are you using tf.math.sqrt() and tf.cast()?
In self.dropout(), are you using training=training?