fix bug for using state.randkey in mutate of the gene

2024-05-31 16:18:02 +08:00
parent 3a7d05f133
commit bc8267bad0
3 changed files with 3 additions and 3 deletions
--- a/tensorneat/algorithm/neat/gene/node/default.py
+++ b/tensorneat/algorithm/neat/gene/node/default.py
@@ -74,7 +74,7 @@ class DefaultNodeGene(BaseNodeGene):
        return jnp.array([bias, res, act, agg])

    def mutate(self, state, randkey, node):
-        k1, k2, k3, k4 = jax.random.split(state.randkey, num=4)
+        k1, k2, k3, k4 = jax.random.split(randkey, num=4)
        index = node[0]

        bias = mutate_float(