File size: 732 Bytes
a15535e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from forgeenv.roles.drift_generator import BaselineDriftGenerator
from forgeenv.roles.prompts import render_drift_generator_prompt
from forgeenv.tasks.task_sampler import TaskSampler
sampler = TaskSampler()
script = sampler.get_by_id("simple_regression").script_content
prompt = render_drift_generator_prompt(script, "ChangeTokenizerBehavior", {"transformers": "4.40"})
fence = "```python"
script_block = ""
if fence in prompt:
script_block = prompt.split(fence, 1)[1].split("```", 1)[0]
print("script_block len:", len(script_block))
print("first 80 chars:", repr(script_block[:80]))
gen = BaselineDriftGenerator(seed=0)
spec = gen.propose(target_category="ChangeTokenizerBehavior", script=script_block)
print("spec:", spec)
|