Input to imagine_ahead in utils.py

[aravindvenu7]

Input should contain only the initial belief and state right? Here, input contains the entire sequence of beliefs and states instead. Not sure how this works with the algorithm