[Spec] Move draft-extend prep to EagleDraftWorkerBase; unify prepare_for_* names (#28093)

This commit is contained in:
Liangsheng Yin
2026-06-12 16:44:04 -07:00
committed by GitHub
parent f23f48df98
commit 87554c7855
12 changed files with 126 additions and 119 deletions
@@ -154,7 +154,7 @@ def _scan_srt():
def _draft_worker_classes():
"""All transitive BaseDraftWorker subclasses under speculative/."""
"""All transitive EagleDraftWorkerBase subclasses under speculative/."""
by_name = {}
for path in sorted(_SPECULATIVE_DIR.glob("*.py")):
rel = path.relative_to(_SRT_DIR).as_posix()
@@ -166,7 +166,7 @@ def _draft_worker_classes():
}
by_name[node.name] = (rel, node, bases)
workers = {"BaseDraftWorker"}
workers = {"EagleDraftWorkerBase"}
changed = True
while changed:
changed = False
@@ -177,7 +177,7 @@ def _draft_worker_classes():
return [
(rel, node)
for name, (rel, node, _) in sorted(by_name.items())
if name in workers and name != "BaseDraftWorker"
if name in workers and name != "EagleDraftWorkerBase"
]