Don't import numba on Python 3.12

scikit-hep · Jan 29, 2024 · 3b2a1a2 · 3b2a1a2
1 parent 6ec6cb0
commit 3b2a1a2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/vector/backends/_numba.py b/src/vector/backends/_numba.py
@@ -5,17 +5,17 @@
 
 from __future__ import annotations
 
+import sys
 import types
 import typing
 
-import numba
+if sys.version_info < (3, 12):
+    import numba
 
 import vector._compute.lorentz
 import vector._compute.planar
 import vector._compute.spatial
 
-__doctest_requires__ = {("*"): ["numba"]}
-
 names_and_modules = [
     ("planar", vector._compute.planar),
     ("spatial", vector._compute.spatial),

diff --git a/src/vector/backends/_numba_object.py b/src/vector/backends/_numba_object.py
@@ -13,8 +13,11 @@
 from __future__ import annotations
 
 import operator
+import sys
+
+if sys.version_info < (3, 12):
+    import numba
 
-import numba
 import numpy
 
 import vector
@@ -50,8 +53,6 @@
     _coord_object_type,
 )
 
-__doctest_requires__ = {("*"): ["numba"]}
-
 
 @numba.extending.overload(numpy.nan_to_num)  # FIXME: This needs to go into Numba!
 def nan_to_num(x, copy=True, nan=0.0, posinf=None, neginf=None):