From 53eeda36ff1b0087559dd771499305c785dd001b Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Fri, 20 Oct 2023 00:59:28 -0700 Subject: [PATCH] Updated jax.config import PiperOrigin-RevId: 575132921 Change-Id: I3c535c9d5f766acd59ed19ff79885deb5d705e42 --- acme/jax/running_statistics_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/acme/jax/running_statistics_test.py b/acme/jax/running_statistics_test.py index 2119515737..64b8105b64 100644 --- a/acme/jax/running_statistics_test.py +++ b/acme/jax/running_statistics_test.py @@ -21,7 +21,7 @@ from acme import specs from acme.jax import running_statistics import jax -from jax.config import config as jax_config +from jax import config as jax_config import jax.numpy as jnp import numpy as np import tree