From ffbc4af21601bbc5a61631f5db9e351fb550b2f5 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Mon, 31 Jul 2023 18:04:30 +0200 Subject: [PATCH] Add vectors registry --- spacy/default_config.cfg | 2 +- spacy/language.py | 2 +- spacy/util.py | 1 + spacy/vectors.pyx | 2 +- 4 files changed, 4 insertions(+), 3 deletions(-) diff --git a/spacy/default_config.cfg b/spacy/default_config.cfg index 812b89165..b005eef40 100644 --- a/spacy/default_config.cfg +++ b/spacy/default_config.cfg @@ -27,7 +27,7 @@ batch_size = 1000 @tokenizers = "spacy.Tokenizer.v1" [nlp.vectors] -@misc = "spacy.Vectors.v1" +@vectors = "spacy.Vectors.v1" # The pipeline components and their models [components] diff --git a/spacy/language.py b/spacy/language.py index a0ccdcbea..26152b90a 100644 --- a/spacy/language.py +++ b/spacy/language.py @@ -1774,7 +1774,7 @@ class Language: # fill in [nlp.vectors] if not present (as a narrower alternative to # auto-filling [nlp] from the default config) if "vectors" not in config["nlp"]: - config["nlp"]["vectors"] = {"@misc": "spacy.Vectors.v1"} + config["nlp"]["vectors"] = {"@vectors": "spacy.Vectors.v1"} config_lang = config["nlp"].get("lang") if config_lang is not None and config_lang != cls.lang: raise ValueError( diff --git a/spacy/util.py b/spacy/util.py index a2a033cbc..1689ac827 100644 --- a/spacy/util.py +++ b/spacy/util.py @@ -118,6 +118,7 @@ class registry(thinc.registry): augmenters = catalogue.create("spacy", "augmenters", entry_points=True) loggers = catalogue.create("spacy", "loggers", entry_points=True) scorers = catalogue.create("spacy", "scorers", entry_points=True) + vectors = catalogue.create("spacy", "vectors", entry_points=True) # These are factories registered via third-party packages and the # spacy_factories entry point. This registry only exists so we can easily # load them via the entry points. The "true" factories are added via the diff --git a/spacy/vectors.pyx b/spacy/vectors.pyx index 919562eeb..6c2131ebb 100644 --- a/spacy/vectors.pyx +++ b/spacy/vectors.pyx @@ -98,7 +98,7 @@ cdef class BaseVectors: return self -@util.registry.misc("spacy.Vectors.v1") +@util.registry.vectors("spacy.Vectors.v1") def create_mode_vectors() -> Callable[["Vocab"], BaseVectors]: def vectors_factory(vocab: "Vocab") -> BaseVectors: return Vectors(strings=vocab.strings)