AssertionError when using nlp.rehearse() with en_core_web_trf

[jama1017]

How to reproduce the behaviour

I want to use rehearsal to correct some wrong dependency parsing tree produced by en_core_web_trf

I used the code directly from: https://github.com/explosion/spaCy/blob/be29216fe2451adead7f56ccd1db494fd8549dae/spacy/tests/training/test_rehearse.py

I modifed test_rehearse to be:

def test_rehearse(component):
    nlp = spacy.load("en_core_web_trf")
    _optimize(nlp, component, REHEARSE_DATA, True)

and called it with test_rehearse("parser")

Your Environment

• spaCy version: 3.8.2
• Platform: Linux-6.8.0-51-generic-x86_64-with-glibc2.39
• Python version: 3.12.7
• Pipelines: en_core_web_sm (3.8.0), en_core_web_lg (3.8.0), en_core_web_trf (3.8.0)

Error Message

I received the following error message: