c7ad5ba6d05809db066b60bc3564201b66d2d51a,tf_agents/drivers/dynamic_episode_driver_test.py,DynamicEpisodeDriverTest,testContinuePreviusRun,#DynamicEpisodeDriverTest#,112

Before Change

        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])
    time_step, policy_state = driver.run()

After Change

    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