C5W3A1 - Translation error

I am getting an error in UNQ_C2 problem:

ValueError: Shape must be rank 2 but is rank 3 for ‘{{node concatenate/concat}} = ConcatV2[N=2, T=DT_FLOAT, Tidx=DT_INT32](bidirectional/concat, repeat_vector/Tile, concatenate/concat/axis)’ with input shapes: [?,64], [?,30,64], .

Getting an error on line: context = one_step_attention(a, s)

This was my for loop:

mentor edit: code removed

In post_activation_LSTM_cell(), your value for initial_state is incorrect. The variables are the 's and ‘c’ from the previous iteration.

Also, please don’t post your code on the forum. That breaks the course Honor Code. I will edit your message to remove the code.