| """Apply pinned RAID surface attacks to both classes, retaining source groups.""" | |
| import argparse,json | |
| from pathlib import Path | |
| from app.robustness import raid_attack | |
| def main(): | |
| p=argparse.ArgumentParser();p.add_argument('input',type=Path);p.add_argument('--output',type=Path,required=True);a=p.parse_args() | |
| with a.output.open('w') as out: | |
| for line in a.input.read_text().splitlines(): | |
| row=json.loads(line) | |
| if row.get('split')!='train':raise ValueError('Augment training partition only; do not leak validation/test derivatives.') | |
| out.write(json.dumps(row|{'attack':'none'})+'\n') | |
| for kind in ['zero_width','whitespace','homoglyph']: | |
| out.write(json.dumps(row|{'text':raid_attack(row['text'],kind)['generation'],'attack':kind})+'\n') | |
| if __name__=='__main__':main() | |