Rllib TensorFlow custom LSTM model giving InvalidArgumentError: Incompatible shapes error (in LSTM layer)

Question

I\'ve been trying to set up a custom LSTM model with RLLib, but for some reason I\'m getting an incompatible shapes error within my LSTM layer when trying to train. In particula