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

Update callbacks.py

Browse Source 
Use ternary
This commit is contained in:
Pierre NAPOLETANO 2023-06-16 11:02:09 +02:00 committed by GitHub
parent e73c1a89bf
commit cb4ae79e6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/ml/callbacks.py
View File

@ -73,10 +73,7 @@ def pipes_with_nvtx_range(
nlp, additional_pipe_functions: Optional[Dict[str, List[str]]] nlp, additional_pipe_functions: Optional[Dict[str, List[str]]]
): ):
for _, pipe in nlp.components: for _, pipe in nlp.components:
if additional_pipe_functions: extra_funcs = additional_pipe_functions.get(pipe.name, []) if additional_pipe_functions else []
extra_funcs = additional_pipe_functions.get(pipe.name, [])
else:
extra_funcs = []
for name in DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS + extra_funcs: for name in DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS + extra_funcs:
func = getattr(pipe, name, None) func = getattr(pipe, name, None)