update test

2025-04-14 10:00:57 +08:00
parent f394280347
commit 638418ccef
1 changed files with 7 additions and 4 deletions
--- a/examples/mujoco_playground/swimmer6.py
+++ b/examples/mujoco_playground/swimmer6.py
@@ -1,13 +1,16 @@
-
 from tensorneat.pipeline import Pipeline
 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
 import jax
+
+
 def random_sample_policy(randkey, obs):
    return jax.random.uniform(randkey, (8,), minval=-1.0, maxval=1.0)
+
+
 if __name__ == "__main__":
    pipeline = Pipeline(
        algorithm=NEAT(