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]