update HyperNEAT;

All example can currently run!

tensorneat/problem/rl_env/brax_env.py

@@ -5,8 +5,8 @@ from .rl_jit import RLEnv
 
 
 class BraxEnv(RLEnv):
-    def __init__(self, env_name: str = "ant", backend: str = "generalized"):
-        super().__init__()
+    def __init__(self, max_step=1000, env_name: str = "ant", backend: str = "generalized"):
+        super().__init__(max_step)
         self.env = envs.create(env_name=env_name, backend=backend)
 
     def env_step(self, randkey, env_state, action):

tensorneat/problem/rl_env/gymnax_env.py

@@ -4,8 +4,8 @@ from .rl_jit import RLEnv
 
 
 class GymNaxEnv(RLEnv):
-    def __init__(self, env_name):
-        super().__init__()
+    def __init__(self, env_name, max_step=1000):
+        super().__init__(max_step)
         assert env_name in gymnax.registered_envs, f"Env {env_name} not registered"
         self.env, self.env_params = gymnax.make(env_name)