add params key into setup.

2024-05-25 17:00:20 +08:00
parent bc378cec7f
commit 5626fddf41
5 changed files with 8 additions and 7 deletions
--- a/tensorneat/algorithm/neat/genome/base.py
+++ b/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):