add save and load function for classes.

This commit is contained in:
wls2002
2024-06-09 20:33:02 +08:00
parent 374c05f5b7
commit 52e5d603f5
13 changed files with 70 additions and 42 deletions

View File

@@ -6,10 +6,10 @@ from algorithm import BaseAlgorithm
from problem import BaseProblem
from problem.rl_env import RLEnv
from problem.func_fit import FuncFit
from utils import State
from utils import State, StatefulBaseClass
class Pipeline:
class Pipeline(StatefulBaseClass):
def __init__(
self,
algorithm: BaseAlgorithm,