W1 A3 E3 predict_and_sample

in ex3 β€œpredict_and_sample” im not sure what is wrong.
According to instructions pred.shape should be (π‘š,𝑇𝑦,π‘›π‘£π‘Žπ‘™π‘’π‘’π‘ ), however i am getting (1,90).
I have used inference_model.predict with a list of the initializers as input.
previous exercises worked ok and I have used np.argmax(pred, axis=-1) for indices

Found the bug,
problem was with previous task (music_inference_model), where I have assigned outputs=out instead of outputs (and still passed the tests)