Ah the imports.

src/pip/_internal/operations/generate_metadata.py

@@ -1,13 +1,18 @@
 """Metadata generation logic for source distributions.
 """
 
+import atexit
 import logging
 import os
 
 from pip._internal.exceptions import InstallationError
 from pip._internal.utils.misc import ensure_dir
 from pip._internal.utils.setuptools_build import make_setuptools_egg_info_args
-from pip._internal.utils.subprocess import call_subprocess
+from pip._internal.utils.subprocess import (
+    call_subprocess,
+    runner_with_spinner_message,
+)
+from pip._internal.utils.temp_dir import TempDirectory
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 from pip._internal.vcs import vcs
 

src/pip/_internal/req/req_install.py

@@ -4,7 +4,6 @@
 
 from __future__ import absolute_import
 
-import atexit
 import logging
 import os
 import shutil