use mp to execute all feature and fix a small bug in config (#157)

This commit is contained in:
Xu Yang
2024-08-03 10:17:01 +08:00
committed by GitHub
parent 38e97dbd81
commit 4eca5fd8c0
5 changed files with 14 additions and 6 deletions
@@ -24,6 +24,7 @@ from rdagent.components.coder.factor_coder.factor import FactorExperiment
from rdagent.core.developer import Developer
from rdagent.core.evolving_agent import RAGEvoAgent
from rdagent.core.scenario import Scenario
from rdagent.log import rdagent_logger as logger
class FactorCoSTEER(Developer[FactorExperiment]):
@@ -107,5 +108,6 @@ class FactorCoSTEER(Developer[FactorExperiment]):
# save new knowledge base
if self.new_knowledge_base_path is not None:
pickle.dump(factor_knowledge_base, open(self.new_knowledge_base_path, "wb"))
logger.info(f"New knowledge base saved to {self.new_knowledge_base_path}")
exp.sub_workspace_list = factor_experiment.sub_workspace_list
return exp
@@ -44,7 +44,7 @@ class FactorImplementSettings(BaseSettings):
max_loop: int = 10
knowledge_base_path: Union[str, None] = None
new_knowledge_base_path: Union[str, None] = knowledge_base_path
new_knowledge_base_path: Union[str, None] = None
python_bin: str = "python"