complete fully stateful!

use black to format all files!
2024-05-26 18:08:43 +08:00
parent cf69b916af
commit 18c3d44c79
41 changed files with 620 additions and 495 deletions
--- a/tensorneat/algorithm/neat/gene/conn/default.py
+++ b/tensorneat/algorithm/neat/gene/conn/default.py
@@ -26,7 +26,7 @@ class DefaultConnGene(BaseConnGene):
        self.weight_replace_rate = weight_replace_rate

    def new_custom_attrs(self, state):
-        return state, jnp.array([self.weight_init_mean])
+        return jnp.array([self.weight_init_mean])

    def new_random_attrs(self, state, randkey):
        weight = (