Remove the deprecated LocalLayout class.

PiperOrigin-RevId: 473310807
google · Dec 16, 2022 · 6066d2f · 6066d2f
1 parent 1077731
commit 6066d2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/balloon_learning_environment/train_acme_qrdqn.py b/balloon_learning_environment/train_acme_qrdqn.py
@@ -19,7 +19,7 @@
 from absl import app
 from absl import flags
 import acme
-from acme.jax.layouts import local_layout
+from acme.jax.deprecated import local_layout
 from acme.utils import counting
 from acme.utils import loggers
 from balloon_learning_environment import acme_utils