From 9db118b2e8cf6af77da1fa6a3092e6d0a1193017 Mon Sep 17 00:00:00 2001 From: natinew77-creator Date: Sun, 7 Dec 2025 18:55:14 -0500 Subject: [PATCH] Fix ImportError: Cannot import experiments from acme.jax The experiments submodule was not exposed in acme/jax/__init__.py, causing 'from acme.jax import experiments' to fail with an ImportError. This import pattern is used in: - The quickstart tutorial (examples/quickstart.ipynb) - All baseline example scripts (rl_continuous, rl_discrete, imitation) - 21+ files throughout the codebase The fix adds the missing import statement to expose the experiments submodule from the acme.jax namespace. Fixes #335 --- acme/jax/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/acme/jax/__init__.py b/acme/jax/__init__.py index 240cb71526..7bbfbb386c 100644 --- a/acme/jax/__init__.py +++ b/acme/jax/__init__.py @@ -12,3 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +"""JAX-specific components for Acme.""" + +# Expose the experiments submodule. +from acme.jax import experiments