fix bugs

2024-07-12 02:25:57 +08:00
parent 3194678a15
commit 5fc63fdaf1
28 changed files with 351 additions and 142 deletions
--- a/examples/gymnax/cartpole.py
+++ b/examples/gymnax/cartpole.py
@@ -5,7 +5,7 @@ from tensorneat.algorithm.neat import NEAT
 from tensorneat.genome import DefaultGenome, BiasNode

 from tensorneat.problem.rl import GymNaxEnv
-from tensorneat.common import Act, Agg
+from tensorneat.common import ACT, AGG



@@ -24,8 +24,8 @@ if __name__ == "__main__":
                num_outputs=2,
                init_hidden_layers=(),
                node_gene=BiasNode(
-                    activation_options=Act.tanh,
-                    aggregation_options=Agg.sum,
+                    activation_options=ACT.tanh,
+                    aggregation_options=AGG.sum,
                ),
                output_transform=jnp.argmax,
            ),