vrr/heretic
2
0
Fork 0
mirror of https://github.com/p-e-w/heretic.git synced 2026-07-09 17:28:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve appearance of selection menus

Browse source
This commit is contained in:
Philipp Emanuel Weidmann 2025-11-16 11:32:58 +05:30
parent 8a1aceff11
commit ea699dce46
No known key found for this signature in database
GPG key ID: CE6A7756AD520E0F
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/heretic/main.py
View file

@ -27,7 +27,7 @@ from optuna.exceptions import ExperimentalWarning
from optuna.samplers import TPESampler
from optuna.study import StudyDirection
from pydantic import ValidationError
from questionary import Choice
from questionary import Choice, Style
from rich.traceback import install
from .config import Settings
@ -315,9 +315,9 @@ def run():
choices = [
Choice(
title=(
f"Trial {trial.user_attrs['index']:>3}: "
f"Refusals {trial.user_attrs['refusals']:>2}/{len(evaluator.bad_prompts)}, "
f"KL divergence {trial.user_attrs['kl_divergence']:.2f}"
f"[Trial {trial.user_attrs['index']:>3}] "
f"Refusals: {trial.user_attrs['refusals']:>2}/{len(evaluator.bad_prompts)}, "
f"KL divergence: {trial.user_attrs['kl_divergence']:.2f}"
),
value=trial,
)
@ -348,6 +348,7 @@ def run():
trial = questionary.select(
"Which trial do you want to use?",
choices=choices,
style=Style([("highlighted", "reverse")]),
).ask()
if trial is None or trial == "":
@ -374,6 +375,7 @@ def run():
"Chat with the model",
"Nothing (return to trial selection menu)",
],
style=Style([("highlighted", "reverse")]),
).ask()
if action is None or action == "Nothing (return to trial selection menu)":
@ -422,6 +424,7 @@ def run():
"Public",
"Private",
],
style=Style([("highlighted", "reverse")]),
).ask()
private = visibility == "Private"