tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update test

Browse Source
This commit is contained in:
Nam-dada
2025-04-14 10:00:57 +08:00
parent f394280347
commit 638418ccef
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/tensorneat/mjx_swimmer.py → examples/mujoco_playground/swimmer6.py
View File

@@ -1,13 +1,16 @@
from tensorneat.pipeline import Pipeline from tensorneat.pipeline import Pipeline
from tensorneat.algorithm.neat import NEAT from tensorneat.algorithm.neat import NEAT
from tensorneat.genome import DefaultGenome, BiasNode, DefaultConn, DefaultMutation from tensorneat.genome import DefaultGenome, BiasNode, DefaultConn, DefaultMutation
from problem.rl import MujocoEnv from tensorneat.problem.rl import MujocoEnv
from tensorneat.common import ACT, AGG from tensorneat.common import ACT, AGG
import jax import jax
def random_sample_policy(randkey, obs): def random_sample_policy(randkey, obs):
return jax.random.uniform(randkey, (8,), minval=-1.0, maxval=1.0) return jax.random.uniform(randkey, (8,), minval=-1.0, maxval=1.0)
if __name__ == "__main__": if __name__ == "__main__":
pipeline = Pipeline( pipeline = Pipeline(
algorithm=NEAT( algorithm=NEAT(