diff --git a/spacy/tokens/doc.pyi b/spacy/tokens/doc.pyi
index 353983fd0..c21a8aa17 100644
--- a/spacy/tokens/doc.pyi
+++ b/spacy/tokens/doc.pyi
@@ -1,20 +1,8 @@
+from typing import Any, Callable, Dict, Iterable, Iterator, List, Optional
+from typing import Protocol, Sequence, Tuple, Union, overload
 from pathlib import Path
-from typing import (
-    Any,
-    Callable,
-    Dict,
-    Iterable,
-    Iterator,
-    List,
-    Optional,
-    Protocol,
-    Sequence,
-    Tuple,
-    Union,
-    overload,
-)
+import numpy
 
-import numpy as np
 from cymem.cymem import Pool
 from thinc.types import Floats1d, Floats2d, Ints2d
 
@@ -40,7 +28,7 @@ class Doc:
     user_hooks: Dict[str, Callable[..., Any]]
     user_token_hooks: Dict[str, Callable[..., Any]]
     user_span_hooks: Dict[str, Callable[..., Any]]
-    tensor: np.ndarray[Any, np.dtype[np.float_]]
+    tensor: numpy.ndarray[Any, numpy.dtype[numpy.float_]]
     user_data: Dict[str, Any]
     has_unknown_spaces: bool
     _context: Any
@@ -164,7 +152,7 @@ class Doc:
     ) -> Doc: ...
     def to_array(
         self, py_attr_ids: Union[int, str, List[Union[int, str]]]
-    ) -> np.ndarray[Any, np.dtype[np.float_]]: ...
+    ) -> numpy.ndarray[Any, numpy.dtype[numpy.float_]]: ...
     @staticmethod
     def from_docs(
         docs: List[Doc],
diff --git a/spacy/training/corpus.py b/spacy/training/corpus.py
index d6695c8e2..63b5a361d 100644
--- a/spacy/training/corpus.py
+++ b/spacy/training/corpus.py
@@ -1,7 +1,7 @@
+from typing import TYPE_CHECKING, Callable, Iterable, List, Optional, Union
 import random
 import warnings
 from pathlib import Path
-from typing import TYPE_CHECKING, Callable, Iterable, List, Optional, Union
 
 import srsly