Skip to content

Commit e1bc7da

Browse files
Make cepo_config required parameter
1 parent 550421f commit e1bc7da

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optillm/cepo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def rate_completions_pairwise(system_prompt: str, initial_query: str, client: An
354354
return completions[best_index], completion_tokens, cb_log
355355

356356

357-
def cepo(system_prompt: str, initial_query: str, client: Any, model: str, cepo_config: Optional[CepoConfig]) -> tuple[str, int]:
357+
def cepo(system_prompt: str, initial_query: str, client: Any, model: str, cepo_config: CepoConfig) -> tuple[str, int]:
358358
"""
359359
Applies CePO reasoning flow for the given task. First, it generates multiple completions, and then rates them to select the best one.
360360
Each completion is generated as follows:

0 commit comments

Comments
 (0)