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/gene/conn/base.py
+++ b/tensorneat/algorithm/neat/gene/conn/base.py
@@ -1,4 +1,4 @@
-import jax, jax.numpy as jnp
+import jax
 from .. import BaseGene