From d887db45be0ede22afcfb4fb02aa4570ec6befe0 Mon Sep 17 00:00:00 2001
From: Matthew Barber <quitesimplymatt@gmail.com>
Date: Wed, 1 Sep 2021 12:13:02 +0100
Subject: Use ep.load() instead of import_module() to get xp

Co-authored-by: Zac Hatfield-Dodds <zac.hatfield.dodds@gmail.com>
---
 numpy/tests/test_public_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'numpy/tests/test_public_api.py')

diff --git a/numpy/tests/test_public_api.py b/numpy/tests/test_public_api.py
index 26402fa2a..1eca25afb 100644
--- a/numpy/tests/test_public_api.py
+++ b/numpy/tests/test_public_api.py
@@ -481,7 +481,7 @@ def test_array_api_entry_point():
     except StopIteration:
         raise AssertionError("'numpy' not in array_api entry points") from None
 
-    xp = importlib.import_module(ep.value)
+    xp = ep.load()
     msg = (
         f"numpy entry point value '{ep.value}' "
         "does not point to our Array API implementation"
-- 
cgit v1.2.1