from generalanalysis.jailbreaks import BijectionLearning, BijectionLearningConfig
from generalanalysis.data_utils import load_harmbench_dataset
config = BijectionLearningConfig(
exp_name="bijection_test",
victim_model="claude-3-7-sonnet-20250219",
trials=20,
bijection_type="digit",
fixed_size=10,
num_digits=2,
safety_data="harmbench",
universal=False,
digit_delimiter=" ",
interleave_practice=False,
context_length_search=False,
prefill=False,
num_teaching_shots=10,
num_multi_turns=0,
input_output_filter=False
)
bijection = BijectionLearning(config)
results = bijection.optimize(load_harmbench_dataset())