c7ad5ba6d05809db066b60bc3564201b66d2d51a,tf_agents/drivers/dynamic_episode_driver_test.py,DynamicEpisodeDriverTest,testOneStepUpdatesObservers,#DynamicEpisodeDriverTest#,133

Before Change

        driver_test_utils.PyEnvironmentMock())
    policy = driver_test_utils.TFPolicyMock(
        env.time_step_spec(), env.action_spec())
    num_episodes_observer = NumEpisodesObserver()
    num_steps_observer = NumStepsObserver()

    driver = dynamic_episode_driver.DynamicEpisodeDriver(
        env, policy, observers=[num_episodes_observer, num_steps_observer])

After Change

        driver_test_utils.PyEnvironmentMock())
    policy = driver_test_utils.TFPolicyMock(
        env.time_step_spec(), env.action_spec())
    num_episodes_observer = driver_test_utils.NumEpisodesObserver()
    num_steps_observer = driver_test_utils.NumStepsObserver()

    driver = dynamic_episode_driver.DynamicEpisodeDriver(
        env, policy, observers=[num_episodes_observer, num_steps_observer])

Instances