explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-11-04 09:57:26 +03:00
Code Issues Packages Projects Releases Wiki Activity

Export function signature to allow introspection of args in tests

Browse Source
This commit is contained in:
shademe 2022-06-14 15:49:01 +02:00
parent d8684f7372
commit f4405ca3ad
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/trainable_pipe.pyx
View File

@ -1,5 +1,6 @@
# cython: infer_types=True, profile=True # cython: infer_types=True, profile=True, binding = True
import functools import functools
import inspect
from typing import Iterable, Iterator, Optional, Dict, Tuple, Callable from typing import Iterable, Iterator, Optional, Dict, Tuple, Callable
import srsly import srsly
from thinc.api import set_dropout_rate, Model, Optimizer from thinc.api import set_dropout_rate, Model, Optimizer
@ -26,6 +27,7 @@ def trainable_pipe_nvtx_range(func):
with use_nvtx_range(f"{self.name} {func.__name__}"): with use_nvtx_range(f"{self.name} {func.__name__}"):
return func(self, *args, **kwargs) return func(self, *args, **kwargs)
inner.__signature__ = inspect.signature(func)
return inner return inner