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/hyperneat/substrate/base.py
+++ b/tensorneat/algorithm/hyperneat/substrate/base.py
@@ -1,9 +1,7 @@
-from utils import State
+from utils import StatefulBaseClass


-class BaseSubstrate:
-    def setup(self, state=State()):
-        return state
+class BaseSubstrate(StatefulBaseClass):

    def make_nodes(self, query_res):
        raise NotImplementedError