add params key into setup.

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

@@ -3,7 +3,7 @@ from utils import State
 
 class BaseCrossover:
 
-    def setup(self, state=State()):
+    def setup(self, key, state=State()):
         return state
 
     def __call__(self, state, key, genome, nodes1, nodes2, conns1, conns2):

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

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

tensorneat/algorithm/neat/gene/base.py

@@ -9,7 +9,7 @@ class BaseGene:
     def __init__(self):
         pass
 
-    def setup(self, state=State()):
+    def setup(self, key, state=State()):
         return state
 
     def new_attrs(self, state, key):

tensorneat/algorithm/neat/genome/base.py

@@ -24,7 +24,7 @@ class BaseGenome:
         self.node_gene = node_gene
         self.conn_gene = conn_gene
 
-    def setup(self, state=State()):
+    def setup(self, key, state=State()):
         return state
 
     def transform(self, state, nodes, conns):

tensorneat/algorithm/neat/species/base.py

@@ -1,8 +1,9 @@
 from utils import State
 
+
 class BaseSpecies:
-    def setup(self, randkey):
-        raise NotImplementedError
+    def setup(self, key, state=State()):
+        return state
 
     def ask(self, state: State):
         raise NotImplementedError
@@ -11,4 +12,4 @@ class BaseSpecies:
         raise NotImplementedError
 
     def speciate(self, state, generation):
-        raise NotImplementedError
+        raise NotImplementedError