From 4a7a60c9e642ec88d584745989b6c697da68550b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= Date: Tue, 31 Jan 2023 12:24:38 +0100 Subject: [PATCH] Rename `pipe_map` to `student_to_teacher` --- spacy/default_config_distillation.cfg | 2 +- spacy/schemas.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/default_config_distillation.cfg b/spacy/default_config_distillation.cfg index 2993f8e48..1926fafa9 100644 --- a/spacy/default_config_distillation.cfg +++ b/spacy/default_config_distillation.cfg @@ -6,7 +6,7 @@ corpus = "corpora.distillation" dropout = 0.1 max_epochs = 1 max_steps = 0 -pipe_map = {} +student_to_teacher = {} [distillation.batcher] @batchers = "spacy.batch_by_words.v1" diff --git a/spacy/schemas.py b/spacy/schemas.py index 51b64580d..e830eb01f 100644 --- a/spacy/schemas.py +++ b/spacy/schemas.py @@ -418,7 +418,7 @@ class ConfigSchemaDistill(BaseModel): max_epochs: StrictInt = Field(..., title="Maximum number of epochs to distill for") max_steps: StrictInt = Field(..., title="Maximum number of steps to distill for") optimizer: Optimizer = Field(..., title="The optimizer to use") - pipe_map: Dict[str, str] = Field(..., title="Mapping from teacher to student pipe") + student_to_teacher: Dict[str, str] = Field(..., title="Mapping from student to teacher pipe") # fmt: on class Config: