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/utils/init.py
+++ b/tensorneat/utils/init.py
@@ -3,3 +3,4 @@ from .aggregation import Agg, agg_func, AGG_ALL
 from .tools import *
 from .graph import *
 from .state import State
+from .stateful_class import StatefulBaseClass