Automated Code Change

PiperOrigin-RevId: 737670804

Author: vizier-team

vizier/_src/benchmarks/experimenters/atari100k_configs/DER.gin

@@ -19,7 +19,7 @@ import dopamine.jax.agents.dqn.dqn_agent
 import dopamine.jax.agents.full_rainbow.full_rainbow_agent
 import dopamine.jax.networks
 import dopamine.labs.atari_100k.atari_100k_rainbow_agent
-import dopamine.replay_memory.prioritized_replay_buffer
+import dopamine.tf.replay_memory.prioritized_replay_buffer
 
 JaxDQNAgent.gamma = 0.99
 # Use 10 instead of 20 as done by SPR paper

vizier/_src/benchmarks/experimenters/atari100k_configs/DrQ.gin

@@ -19,7 +19,7 @@ import dopamine.jax.agents.dqn.dqn_agent
 import dopamine.jax.agents.full_rainbow.full_rainbow_agent
 import dopamine.jax.networks
 import dopamine.labs.atari_100k.atari_100k_rainbow_agent
-import dopamine.replay_memory.prioritized_replay_buffer
+import dopamine.tf.replay_memory.prioritized_replay_buffer
 
 # Parameters specific to DrQ are higlighted by comments
 JaxDQNAgent.gamma = 0.99

vizier/_src/benchmarks/experimenters/atari100k_configs/DrQ_eps.gin

@@ -19,7 +19,7 @@ import dopamine.jax.agents.dqn.dqn_agent
 import dopamine.jax.agents.full_rainbow.full_rainbow_agent
 import dopamine.jax.networks
 import dopamine.labs.atari_100k.atari_100k_rainbow_agent
-import dopamine.replay_memory.prioritized_replay_buffer
+import dopamine.tf.replay_memory.prioritized_replay_buffer
 
 # Parameters specific to DrQ are higlighted by comments
 JaxDQNAgent.gamma = 0.99

vizier/_src/benchmarks/experimenters/atari100k_configs/OTRainbow.gin

@@ -19,7 +19,7 @@ import dopamine.jax.agents.dqn.dqn_agent
 import dopamine.jax.agents.full_rainbow.full_rainbow_agent
 import dopamine.jax.networks
 import dopamine.labs.atari_100k.atari_100k_rainbow_agent
-import dopamine.replay_memory.prioritized_replay_buffer
+import dopamine.tf.replay_memory.prioritized_replay_buffer
 
 # Parameters specific to OTRainbow are higlighted by comments
 JaxDQNAgent.gamma = 0.99