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/genome/base.py
+++ b/tensorneat/algorithm/neat/genome/base.py
@@ -1,10 +1,10 @@
 import jax, jax.numpy as jnp
 from ..gene import BaseNodeGene, BaseConnGene
 from ..ga import BaseMutation, BaseCrossover
-from utils import State
+from utils import State, StatefulBaseClass


-class BaseGenome:
+class BaseGenome(StatefulBaseClass):
    network_type = None

    def __init__(