Merge pull request #122 from AngelFP/fix_numba_2

Raise `RuntimeError` for numba 0.57.0
AngelFP · May 15, 2023 · 0083148 · 0083148
2 parents e3acc82 + 9333f85
commit 0083148
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/wake_t/__init__.py b/wake_t/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '0.6.0'
+__version__ = '0.6.1'
 
 
 from .beamline_elements import (PlasmaStage, PlasmaRamp, ActivePlasmaLens,

diff --git a/wake_t/utilities/numba.py b/wake_t/utilities/numba.py
@@ -1,7 +1,14 @@
 """This module contains custom definitions of the numba decorators."""
 
 import os
-from numba import njit
+from numba import njit, __version__ as numba_version
+
+
+if numba_version == '0.57.0':
+    raise RuntimeError(
+        'Wake-T is incompatible with numba 0.57.0.\n'
+        'Please install either a later or an earlier version.'
+    )
 
 
 # Check if the environment variable WAKET_DISABLE_CACHING is set to 1