Gridworld.draw should work seamlessly with POMDPs. Ideally it would take a POMDP as the world argument. For the trajectory argument, it should take mdp states or mdp state-action pairs, and it should take pomdp states or pomdp state-action pairs. (this means dealing with a few different cases, but i think mdp, pomdp and state/state-action is enought. if it works for those, it convers almost every case we need).
Gridworld.draw should work seamlessly with POMDPs. Ideally it would take a POMDP as the world argument. For the
trajectoryargument, it should take mdp states or mdp state-action pairs, and it should take pomdp states or pomdp state-action pairs. (this means dealing with a few different cases, but i think mdp, pomdp and state/state-action is enought. if it works for those, it convers almost every case we need).