GymnaxtoBrax Wrapper imports failing
limbryan opened this issue · comments
Hey!
gymnax/gymnax/wrappers/brax.py
Line 9 in 9155a96
This line seems to be throwing an error when using the wrapper. Looking at the other wrappers, should probably be something like this?
from gymnax.environments.environment import Environment, EnvState, EnvParams