Skip to content

Commit eeea270

Browse files
committed
fix prompt type in helper
1 parent be16e15 commit eeea270

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

promptolution/helpers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,11 +73,12 @@ def run_optimization(df: pd.DataFrame, config: "ExperimentConfig") -> List[Promp
7373
predictor = get_predictor(llm, config=config)
7474

7575
if getattr(config, "prompts") is None:
76-
config.prompts = create_prompts_from_task_description(
76+
initial_prompts = create_prompts_from_task_description(
7777
task_description=config.task_description,
7878
llm=llm,
7979
n_prompts=config.n_initial_prompts,
8080
)
81+
config.prompts = [Prompt(p) for p in initial_prompts]
8182

8283
if config.optimizer == "capo" and (config.eval_strategy is None or "block" not in config.eval_strategy):
8384
logger.warning("📌 CAPO requires block evaluation strategy. Setting it to 'sequential_block'.")

0 commit comments

Comments
 (0)