Python bindings: setup.py: improve numpy fixing (fixes #6700)

OSGeo · Nov 16, 2022 · 9f534e7 · 9f534e7
1 parent b77481c
commit 9f534e7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/swig/python/setup.py.in b/swig/python/setup.py.in
@@ -60,7 +60,14 @@ libraries = ['@GDAL_LIB_OUTPUT_NAME@']
 # Function to find numpy's include directory
 def get_numpy_include():
     # Fix numpy installation using setuptools
-    __builtins__.__NUMPY_SETUP__ = False
+    # __builtins__ can be either a dict or a module
+    def _set_builtin(name, value):
+        if isinstance(__builtins__, dict):
+            __builtins__[name] = value
+        else:
+            setattr(__builtins__, name, value)
+
+    _set_builtin("__NUMPY_SETUP__", False)
 
     import numpy
     return numpy.get_include()