add obs normalization for rl env

2024-06-14 16:11:50 +08:00
parent aac9f4c3fb
commit b9d6482d11
3 changed files with 135 additions and 19 deletions
--- a/tensorneat/algorithm/neat/genome/default.py
+++ b/tensorneat/algorithm/neat/genome/default.py
@@ -270,7 +270,7 @@ class DefaultGenome(BaseGenome):

            fixed_args_output_funcs.append(f)

-        forward_func = lambda inputs: [f(inputs) for f in fixed_args_output_funcs]
+        forward_func = lambda inputs: jnp.array([f(inputs) for f in fixed_args_output_funcs])

        return (
            symbols,