All function with state will update the state and return it.

Remove randkey args in functions with state, since it can attach the randkey by states.

tensorneat/algorithm/neat/ga/mutation/base.py

@@ -3,8 +3,8 @@ from utils import State
 
 class BaseMutation:
 
-    def setup(self, key, state=State()):
+    def setup(self, state=State()):
         return state
 
-    def __call__(self, state, key, genome, nodes, conns, new_node_key):
+    def __call__(self, state, genome, nodes, conns, new_node_key):
         raise NotImplementedError