able_rl.foundation.agent.Agent.predict_rewards">[docs]</a>    <span class="k">def</span> <span class="nf">predict_rewards</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">states</span><span class="p">,</span> <span class="n">actions</span><span class="p">):</span>
<span class="w">        </span><span class="sd">&quot;&quot;&quot;Predict reward for a list of state-actions.</span>

<span class="sd">        This function uses the avg reward matrix (which simula