numpy
diff --git a/‎numpy/testing/_private/utils.py
Copy file name to clipboardExpand all lines: numpy/testing/_private/utils.py
+5-4Lines changed: 5 additions & 4 deletions b/‎numpy/testing/_private/utils.py
Copy file name to clipboardExpand all lines: numpy/testing/_private/utils.py
+5-4Lines changed: 5 additions & 4 deletions
@@ -62,9 +62,12 @@ class KnownFailureException(Exception):
 KnownFailureTest = KnownFailureException  # backwards compat
 verbose = 0
 
+NUMPY_ROOT = pathlib.Path(np.__file__).parent
+
 try:
-    np_dist = importlib_metadata.distribution('numpy')
-except importlib_metadata.PackageNotFoundError:
+    # Make sure we find the distibution in the path where the module was loaded from.
+    np_dist = next(importlib_metadata.distributions(name='numpy', path=[NUMPY_ROOT.parent]))
+except StopIteration:
     IS_INSTALLED = IS_EDITABLE = False
 else:
     IS_INSTALLED = True
@@ -90,8 +93,6 @@ class KnownFailureException(Exception):
 
 NOGIL_BUILD = bool(sysconfig.get_config_var("Py_GIL_DISABLED"))
 
-NUMPY_ROOT = pathlib.Path(np.__file__).parent
-
 def assert_(val, msg=''):
     """
     Assert that works in release mode.