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/species/base.py
+++ b/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