fix: rename registry.py to fix class finding when pickling

aphp · Jan 5, 2024 · 6d842c7 · 6d842c7
1 parent 49539c0
commit 6d842c7
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/edsnlp/core/__init__.py b/edsnlp/core/__init__.py
@@ -1,2 +1,2 @@
-from .registry import registry
+from .registries import registry
 from .pipeline import PipelineProtocol
diff --git a/edsnlp/core/registry.py → edsnlp/core/registries.py b/edsnlp/core/registry.py → edsnlp/core/registries.py
diff --git a/edsnlp/patch_spacy.py b/edsnlp/patch_spacy.py
@@ -7,7 +7,7 @@
 from spacy.util import SimpleFrozenDict, SimpleFrozenList, raise_error
 from spacy.vocab import create_vocab
 
-from edsnlp.core.registry import accepted_arguments
+from edsnlp.core.registries import accepted_arguments
 
 
 @classmethod

diff --git a/edsnlp/train.py b/edsnlp/train.py
@@ -31,7 +31,7 @@
 
 import edsnlp
 from edsnlp.core.pipeline import Pipeline
-from edsnlp.core.registry import registry
+from edsnlp.core.registries import registry
 from edsnlp.optimization import LinearSchedule, ScheduledOptimizer
 from edsnlp.pipes.trainable.embeddings.transformer.transformer import Transformer
 from edsnlp.scorers import Scorer

diff --git a/tests/training/test_train.py b/tests/training/test_train.py
@@ -10,7 +10,7 @@
 from confit.utils.random import set_seed
 from spacy.tokens import Span
 
-from edsnlp.core.registry import registry
+from edsnlp.core.registries import registry
 from edsnlp.data.converters import AttributesMappingArg, get_current_tokenizer
 from edsnlp.train import GenericScorer, Reader, train
 from edsnlp.utils.span_getters import SpanSetterArg, set_spans