(env, policy, gamma = 1.0, render = False)
return total_reward
Runs an episode and return the total reward
Runs an episode and return the total reward
Runs an episode and return the total reward
def run_episode(env, policy, gamma = 1.0, render = False):
    """ Runs an episode and return the total reward """
    obs = env.reset()
    total_reward = 0
    step_idx = 0
    while True:
        if render:
            env.render()
        obs, reward, done , _ = env.step(int(policy[obs]))
        total_reward += (gamma ** step_idx * reward)
        step_idx += 1
        if done:
            break
    return total_reward
def run_episode ( env , policy , gamma = 1.0 , render = False ) : obs = env . reset ( ) total_reward = 0 step_idx = 0 while True : if render : env . render ( ) obs , reward , done , _ = env . step ( int ( policy [ obs ] ) ) total_reward += ( gamma ** step_idx * reward ) step_idx += 1 if done : break return total_reward