Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from medcat.components.linking import context_based_linker
from medcat.components.types import _DEFAULT_LINKING as DEF_LINKING
from medcat.components import types
from medcat.config import Config
from medcat.vocab import Vocab
Expand Down Expand Up @@ -35,7 +36,7 @@ def weighted_average_function(self, nr: int) -> float:


class LinkingInitTests(ComponentInitTests, unittest.TestCase):
expected_def_components = 3
expected_def_components = len(DEF_LINKING)
comp_type = types.CoreComponentType.linking
default_cls = context_based_linker.Linker
default_creator = context_based_linker.Linker.create_new_component
Expand Down
2 changes: 1 addition & 1 deletion medcat-v2/tests/components/ner/test_vocab_based_ner.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def __init__(self, config: Config):


class NerInitTests(ComponentInitTests, unittest.TestCase):
expected_def_components = 3
expected_def_components = len(types._DEFAULT_NER)
comp_type = types.CoreComponentType.ner
default_cls = vocab_based_ner.NER
default_creator = vocab_based_ner.NER.create_new_component
Expand Down
Loading