From cb4ae79e6aa6a31fe906a3432ccbb923b0ac8144 Mon Sep 17 00:00:00 2001 From: Pierre NAPOLETANO <56818795+PierreNAPOLETANO@users.noreply.github.com> Date: Fri, 16 Jun 2023 11:02:09 +0200 Subject: [PATCH] Update callbacks.py Use ternary --- spacy/ml/callbacks.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/spacy/ml/callbacks.py b/spacy/ml/callbacks.py index e2378a7ba..db4a46b2d 100644 --- a/spacy/ml/callbacks.py +++ b/spacy/ml/callbacks.py @@ -73,10 +73,7 @@ def pipes_with_nvtx_range( nlp, additional_pipe_functions: Optional[Dict[str, List[str]]] ): for _, pipe in nlp.components: - if additional_pipe_functions: - extra_funcs = additional_pipe_functions.get(pipe.name, []) - else: - extra_funcs = [] + extra_funcs = additional_pipe_functions.get(pipe.name, []) if additional_pipe_functions else [] for name in DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS + extra_funcs: func = getattr(pipe, name, None)