Defer numpy import on first pass of setup.py

AIM-Harvard · May 22, 2023 · 00c8ec5 · 00c8ec5
1 parent 6a761c4
commit 00c8ec5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,13 @@
 from distutils import sysconfig
 import platform
 
-import numpy
+try:
+ import numpy
+except: # noqa: E722
+ from unittest.mock import MagicMock
+ sys.modules["numpy"] = MagicMock()
+ import numpy
+
 from setuptools import Extension, setup
 import versioneer