add save and load function for classes.

2024-06-09 20:33:02 +08:00
parent 374c05f5b7
commit 52e5d603f5
13 changed files with 70 additions and 42 deletions
--- a/tensorneat/algorithm/neat/species/base.py
+++ b/tensorneat/algorithm/neat/species/base.py
@@ -1,15 +1,12 @@
-from utils import State
+from utils import State, StatefulBaseClass
 from ..genome import BaseGenome


-class BaseSpecies:
+class BaseSpecies(StatefulBaseClass):
    genome: BaseGenome
    pop_size: int
    species_size: int

-    def setup(self, state=State()):
-        return state
-
    def ask(self, state: State):
        raise NotImplementedError