fix: Cache all possible predictions in cache_predictions

fix: Cache all possible predictions in cache_predictions #41