From 5fdf33ea14057f29ca3becbfed4b7b8f692aabd9 Mon Sep 17 00:00:00 2001 From: Djordje Grbic Date: Tue, 16 Apr 2024 11:18:15 +0200 Subject: [PATCH 1/2] Fix env_name typo --- tensorneat/examples/brax/half_cheetah.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorneat/examples/brax/half_cheetah.py b/tensorneat/examples/brax/half_cheetah.py index da4823c..4d31efe 100644 --- a/tensorneat/examples/brax/half_cheetah.py +++ b/tensorneat/examples/brax/half_cheetah.py @@ -23,7 +23,7 @@ if __name__ == '__main__': ), ), problem=BraxEnv( - env_name='halhcheetah', + env_name='halfcheetah', ), generation_limit=10000, fitness_target=5000 From 4336c81ae347e7baf6d29009d7e6ba8896d7bd0e Mon Sep 17 00:00:00 2001 From: Djordje Grbic Date: Tue, 16 Apr 2024 12:18:22 +0200 Subject: [PATCH 2/2] Remove `state` from `act_func` arguments --- tensorneat/problem/rl_env/brax_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorneat/problem/rl_env/brax_env.py b/tensorneat/problem/rl_env/brax_env.py index 9c34501..4e0b505 100644 --- a/tensorneat/problem/rl_env/brax_env.py +++ b/tensorneat/problem/rl_env/brax_env.py @@ -39,7 +39,7 @@ class BraxEnv(RLEnv): def step(key, env_state, obs): key, _ = jax.random.split(key) - action = act_func(state, obs, params) + action = act_func(obs, params) obs, env_state, r, done, _ = self.step(randkey, env_state, action) return key, env_state, obs, r, done