I fixed most of the typing issues under pylint --strict.
1.1 This required unifying the interface of step, step_env, get_obs, 'is_terminal` for proper inheritance.
spaces now depends on gymnasium instead of gym, adding an additional dependency. Maybe later on gym dependency can be completely removed since it's not maintained anymore? haven't looked into that.
I removed _DeviceArray from the tests as it's deprecated. I use jax version 0.4.24
I ran everything in python 3.10 so technically you could support that too now.
In wrapper/brax.py, GymnaxtoBraxWrapper has two new methods backend and observation_size, as its required to be defined under newest brax version. I used brax version 0.10.0.
I fixed most of the typing issues under pylint --strict. 1.1 This required unifying the interface of
step
,step_env
,get_obs
, 'is_terminal` for proper inheritance.spaces
now depends ongymnasium
instead ofgym
, adding an additional dependency. Maybe later ongym
dependency can be completely removed since it's not maintained anymore? haven't looked into that.I removed
_DeviceArray
from the tests as it's deprecated. I use jax version0.4.24
I ran everything in
python 3.10
so technically you could support that too now.In
wrapper/brax.py
,GymnaxtoBraxWrapper
has two new methodsbackend
andobservation_size
, as its required to be defined under newest brax version. I used brax version0.10.0
.