fix imports brax physics module

adaptive-intelligent-robotics · Nov 30, 2023 · 965573f · 965573f
1 parent 19ce124
commit 965573f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/environments_test/wrapper_test.py b/tests/environments_test/wrapper_test.py
@@ -4,9 +4,9 @@
 import jax
 import jax.numpy as jnp
 import pytest
-from brax.physics.base import vec_to_arr
-from brax.physics.config_pb2 import Joint
 from brax.v1 import jumpy as jp
+from brax.v1.physics.base import vec_to_arr
+from brax.v1.physics.config_pb2 import Joint
 
 from qdax import environments