Fix warning about deprecated module: imp (#6362)

This commit is contained in:
Neil Girdhar
2021-06-17 16:02:22 -04:00
committed by GitHub
parent c58ae94225
commit 8fd7861b75

View File

@@ -16,7 +16,6 @@
compatibility helpers for numpy. """ compatibility helpers for numpy. """
import sys import sys
import imp
PY2 = sys.version_info[0] == 2 PY2 = sys.version_info[0] == 2
PY26 = sys.version_info[0:2] == (2, 6) PY26 = sys.version_info[0:2] == (2, 6)
@@ -26,12 +25,14 @@ PY3 = sys.version_info[0] == 3
PY34 = sys.version_info[0:2] >= (3, 4) PY34 = sys.version_info[0:2] >= (3, 4)
if PY3: if PY3:
import importlib.machinery
string_types = (str,) string_types = (str,)
binary_types = (bytes,bytearray) binary_types = (bytes,bytearray)
range_func = range range_func = range
memoryview_type = memoryview memoryview_type = memoryview
struct_bool_decl = "?" struct_bool_decl = "?"
else: else:
import imp
string_types = (unicode,) string_types = (unicode,)
if PY26 or PY27: if PY26 or PY27:
binary_types = (str,bytearray) binary_types = (str,bytearray)
@@ -52,6 +53,10 @@ def import_numpy():
Returns the numpy module if it exists on the system, Returns the numpy module if it exists on the system,
otherwise returns None. otherwise returns None.
""" """
if PY3:
numpy_exists = (
importlib.machinery.PathFinder.find_spec('numpy') is not None)
else:
try: try:
imp.find_module('numpy') imp.find_module('numpy')
numpy_exists = True numpy_exists = True