I was getting a different error in this block despite passing local tests earlier. Got it to pass by playing around with the syntax of the second argument in the reshape function as suggested by this comment. I’m not sure what the problem was. My implementation did not have hardcoded dimensions to deal with the local example only, so I found it odd. Anyways, hopefully this helps.
Lucas