Add bypass distillation (blockwise local KD) to puzzletron pipeline#1111

Open

Separius wants to merge 16 commits intomainfrom

ssameni/puzzletron-bypass

Commits on May 6, 2026

Add bypass distillation stage to puzzletron pipeline
Separius
committed
Fix mypy attr-defined errors from top-level datasets imports
Separius
committed
Address CodeRabbit review and revert non-bypass datasets-import workarounds
Separius
committed
Remove unused factory aliases; reflow source_datasets_to_discard kwargs
Separius
committed
Add bypass test coverage and address remaining review findings
Separius
committed
Fix CI: F841, mypy attr-defined re-export, ruff format reflows
Separius
committed
Fix CI: ruff format remaining _setup_hydra_cfg_and_pruning call sites
Separius
committed
Merge branch 'main' into ssameni/puzzletron-bypass
Separius
authored

Commits on May 7, 2026