Faster import (lazy numpy)

anivegesana · Aug 8, 2023 · 153eb03 · 153eb03
1 parent 17b96b1
commit 153eb03
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/simple_parsing/helpers/hparams/priors.py b/simple_parsing/helpers/hparams/priors.py
@@ -14,13 +14,16 @@
     overload,
 )
 
-numpy_installed = False
-try:
-    import numpy as np
+import importlib.util
 
-    numpy_installed = True
-except ImportError:
-    pass
+class _np_lazy:
+    def __getattr__(self, attr):
+        global np
+        import numpy as np
+        return getattr(np, attr)
+
+np = _np_lazy()
+numpy_installed = importlib.util.find_spec("numpy") is not None
 
 
 T = TypeVar("T")