From b0b115ff394f381e18c53e7d99e69390e03aaf4f Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Fri, 22 Oct 2021 13:03:10 +0200 Subject: [PATCH] Auto-format code with black (#9530) Co-authored-by: explosion-bot --- spacy/ml/__init__.py | 2 +- spacy/ml/callbacks.py | 4 +++- spacy/tests/pipeline/test_entity_linker.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/spacy/ml/__init__.py b/spacy/ml/__init__.py index d44c0f850..fce8ae5af 100644 --- a/spacy/ml/__init__.py +++ b/spacy/ml/__init__.py @@ -1,2 +1,2 @@ -from .callbacks import create_models_with_nvtx_range # noqa: F401 +from .callbacks import create_models_with_nvtx_range # noqa: F401 from .models import * # noqa: F401, F403 diff --git a/spacy/ml/callbacks.py b/spacy/ml/callbacks.py index c09f911b7..b0d088182 100644 --- a/spacy/ml/callbacks.py +++ b/spacy/ml/callbacks.py @@ -30,7 +30,9 @@ def create_models_with_nvtx_range( ) for node in models.walk(): - with_nvtx_range(node, forward_color=forward_color, backprop_color=backprop_color) + with_nvtx_range( + node, forward_color=forward_color, backprop_color=backprop_color + ) return nlp diff --git a/spacy/tests/pipeline/test_entity_linker.py b/spacy/tests/pipeline/test_entity_linker.py index 247443489..a98d01964 100644 --- a/spacy/tests/pipeline/test_entity_linker.py +++ b/spacy/tests/pipeline/test_entity_linker.py @@ -168,7 +168,7 @@ def test_kb_serialize_2(nlp): def test_kb_set_entities(nlp): - """ Test that set_entities entirely overwrites the previous set of entities """ + """Test that set_entities entirely overwrites the previous set of entities""" v = [5, 6, 7, 8] v1 = [1, 1, 1, 0] v2 = [2, 2, 2, 3]