spaCy/spacy
Madeesh Kannan eaf66e7431
Add NVTX ranges to `TrainablePipe` components (#10965)
* `TrainablePipe`: Add NVTX range decorator

* Annotate `TrainablePipe` subclasses with NVTX ranges

* Export function signature to allow introspection of args in tests

* Revert "Annotate `TrainablePipe` subclasses with NVTX ranges"

This reverts commit d8684f7372.

* Revert "Export function signature to allow introspection of args in tests"

This reverts commit f4405ca3ad.

* Revert "`TrainablePipe`: Add NVTX range decorator"

This reverts commit 26536eb6b8.

* Add `spacy.pipes_with_nvtx_range` pipeline callback

* Show warnings for all missing user-defined pipe functions that need to be annotated
Fix imports, typos

* Rename `DEFAULT_ANNOTATABLE_PIPE_METHODS` to `DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS`
Reorder import

* Walk model nodes directly whilst applying NVTX ranges
Ignore pipe method wrapper when applying range
2022-06-30 11:28:12 +02:00
..
cli Try cloning repo from main & master (#10843) 2022-06-28 09:11:15 -04:00
displacy fix to horizontal space (#10994) 2022-06-28 20:42:40 +02:00
lang Handle Cyrillic combining diacritics (#10837) 2022-06-28 15:35:32 +02:00
matcher Addition of min_max quantifier in matcher {n,m} (#10981) 2022-06-30 11:01:58 +02:00
ml Add NVTX ranges to `TrainablePipe` components (#10965) 2022-06-30 11:28:12 +02:00
pipeline account for NER labels with a hyphen in the name (#10960) 2022-06-17 20:02:37 +01:00
tests Revert "disable failing test because Stanford servers are down (#11015)" (#11054) 2022-06-30 11:24:54 +02:00
tokens Update vector handling in similarity methods (#11013) 2022-06-28 19:50:47 +02:00
training `Example.get_aligned_parse`: Handle unit and zero length vectors correctly (#11026) 2022-06-28 19:42:58 +02:00
__init__.pxd
__init__.py `enable` argument for spacy.load() (#10784) 2022-06-17 20:24:13 +01:00
__main__.py
about.py Set version to v3.4.0 2022-06-24 08:48:58 +02:00
attrs.pxd
attrs.pyx Intify IOB (#9738) 2022-01-20 13:19:38 +01:00
compat.py Custom component types in spacy.ty (#9469) 2021-10-21 15:31:06 +02:00
default_config.cfg Add a few docs to the default_config.cfg (#9981) 2022-01-05 09:16:40 +01:00
default_config_pretraining.cfg Add new parameter for saving every n epoch in pretraining (#8912) 2021-08-12 11:14:48 +02:00
errors.py Add NVTX ranges to `TrainablePipe` components (#10965) 2022-06-30 11:28:12 +02:00
glossary.py Add glossary entry for root (#10821) 2022-05-20 09:56:32 +02:00
kb.pxd
kb.pyx Made _initialize_X() methods private. (#10978) 2022-06-17 15:55:34 +02:00
language.py `enable` argument for spacy.load() (#10784) 2022-06-17 20:24:13 +01:00
lexeme.pxd
lexeme.pyi fix type of lexeme.rank (#9979) 2022-01-04 13:15:25 +01:00
lexeme.pyx Bugfix for similarity return types (#10051) 2022-01-20 11:40:46 +01:00
lookups.py Fix issues for Mypy 0.950 and Pydantic 1.9.0 (#10786) 2022-05-25 09:33:54 +02:00
morphology.pxd Clean up Morphology imports and definitions (#7441) 2021-04-26 16:54:23 +02:00
morphology.pyx Clean up Morphology imports and definitions (#7441) 2021-04-26 16:54:23 +02:00
parts_of_speech.pxd
parts_of_speech.pyx
pipe_analysis.py 🏷 Add Mypy check to CI and ignore all existing Mypy errors (#9167) 2021-10-14 15:21:40 +02:00
py.typed
schemas.py Addition of min_max quantifier in matcher {n,m} (#10981) 2022-06-30 11:01:58 +02:00
scorer.py Alignment: use a simplified ragged type for performance (#10319) 2022-04-01 09:02:06 +02:00
strings.pxd Update Cython string types (#9143) 2021-09-13 17:02:17 +02:00
strings.pyi Fix StringStore.__getitem__ return type depending on parameter types (#10741) 2022-05-03 17:57:07 +02:00
strings.pyx Update Cython string types (#9143) 2021-09-13 17:02:17 +02:00
structs.pxd
symbols.pxd
symbols.pyx
tokenizer.pxd Add tokenizer option to allow Matcher handling for all rules (#10452) 2022-03-24 13:21:32 +01:00
tokenizer.pyx Add tokenizer option to allow Matcher handling for all rules (#10452) 2022-03-24 13:21:32 +01:00
ty.py Custom component types in spacy.ty (#9469) 2021-10-21 15:31:06 +02:00
typedefs.pxd
typedefs.pyx
util.py Backport parser/alignment optimizations from `feature/refactor-parser` (#10952) 2022-06-24 13:39:52 +02:00
vectors.pyx vectors: avoid expensive comparisons between numpy ints and Python ints (#10992) 2022-06-29 12:58:31 +02:00
vocab.pxd Add support for floret vectors (#8909) 2021-10-27 14:08:31 +02:00
vocab.pyi Add vector deduplication (#10551) 2022-03-30 08:54:23 +02:00
vocab.pyx Add vector deduplication (#10551) 2022-03-30 08:54:23 +02:00