Jazz Improvisation: Test for djmodel summary fails

I don’t want to unnecessarily upload my code, but I have tried slicing X with regular slicing like in np and also with a Keras Lambda layer, but unfortunately I always get the attached error. Everything else works and in the end I can play the generated jazz snippet, but I have no idea how to get past the grader with this. Thanks for your help!

Send me your code in a private message. Click my name and message.