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/ga/mutation/base.py
+++ b/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):