Skip to content

Commit 5d541ee

Browse files
committed
Update optillm.py
fix CLI args
1 parent 95cc14d commit 5d541ee

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optillm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -389,9 +389,9 @@ def parse_args():
389389
args_env = [
390390
("--optillm-api-key", "OPTILLM_API_KEY", str, "", "Optional API key for client authentication to optillm"),
391391
("--approach", "OPTILLM_APPROACH", str, "auto", "Inference approach to use", known_approaches),
392-
("--simulations", "OPTILLM_SIMULATIONS", int, 2, "Number of MCTS simulations"),
393-
("--exploration", "OPTILLM_EXPLORATION", float, 0.2, "Exploration weight for MCTS"),
394-
("--depth", "OPTILLM_DEPTH", int, 1, "Simulation depth for MCTS"),
392+
("--mcts-simulations", "OPTILLM_SIMULATIONS", int, 2, "Number of MCTS simulations"),
393+
("--mcts-exploration", "OPTILLM_EXPLORATION", float, 0.2, "Exploration weight for MCTS"),
394+
("--mcts-depth", "OPTILLM_DEPTH", int, 1, "Simulation depth for MCTS"),
395395
("--model", "OPTILLM_MODEL", str, "gpt-4o-mini", "OpenAI model to use"),
396396
("--rstar-max-depth", "OPTILLM_RSTAR_MAX_DEPTH", int, 3, "Maximum depth for rStar algorithm"),
397397
("--rstar-num-rollouts", "OPTILLM_RSTAR_NUM_ROLLOUTS", int, 5, "Number of rollouts for rStar algorithm"),

0 commit comments

Comments
 (0)