Revert "removing unnecessary call to compute prediction"

This reverts commit 5e116d4e9040475dabc44171c3a972c0a4164862.

reverting removal of compute prediction

necessary for multi-step problems where the update is called on the
previous action set