`act_func` call in `problem/rl_env/brax_env.py:42` is causing an error.

EMI-Group / tensorneat

GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)

BSD 3-Clause "New" or "Revised" License

56 stars 11 forks source link

`act_func` call in `problem/rl_env/brax_env.py:42` is causing an error. #2

Closed djole closed 4 months ago

djole commented 5 months ago

Hi, when calling pipeline.show on a Brax environment, there is an error complaining that we're giving it 4 arguments when 3 are requested. I think the state shouldn't be passed to the act_func function.

Best, Djordje

WLS2002 commented 4 months ago

Thank you very much for taking the time to identify this issue! I'll merge your pull request for solving this issue. Thanks a lot!