simplify numba import

Author: Jammy2211

autoarray/inversion/inversion/imaging/w_tilde.py

@@ -53,7 +53,7 @@ def __init__(
             import numba
         except ModuleNotFoundError:
             raise exc.InversionException(
-                "Inversion functionality (linear light profiles, pixelized reconstructions) is "
+                "Inversion w-tilde functionality (pixelized reconstructions) is "
                 "disabled if numba is not installed.\n\n"
                 "This is because the run-times without numba are too slow.\n\n"
                 "Please install numba, which is described at the following web page:\n\n"

autoarray/inversion/inversion/interferometer/w_tilde.py

@@ -53,6 +53,17 @@ def __init__(
             the simultaneous linear equations are combined and solved simultaneously.
         """
 
+        try:
+            import numba
+        except ModuleNotFoundError:
+            raise exc.InversionException(
+                "Inversion w-tilde functionality (pixelized reconstructions) is "
+                "disabled if numba is not installed.\n\n"
+                "This is because the run-times without numba are too slow.\n\n"
+                "Please install numba, which is described at the following web page:\n\n"
+                "https://pyautolens.readthedocs.io/en/latest/installation/overview.html"
+            )
+
         self.w_tilde = w_tilde
         self.w_tilde.check_noise_map(noise_map=dataset.noise_map)
 

autoarray/numba_util.py

@@ -1,27 +1,10 @@
-import os
-from functools import wraps
 import logging
-import time
-from typing import Callable
 
 from autoconf import conf
 
-from autoarray import exc
 
 logger = logging.getLogger(__name__)
 
-"""
-Depending on if we're using a super computer, we want two different numba decorators:
-
-If on laptop:
-
-@numba.jit(nopython=True, cache=True, parallel=False)
-
-If on super computer:
-
-@numba.jit(nopython=True, cache=False, parallel=True)
-"""
-
 try:
     nopython = conf.instance["general"]["numba"]["nopython"]
     cache = conf.instance["general"]["numba"]["cache"]
@@ -31,41 +14,19 @@
     cache = True
     parallel = False
 
-try:
-    if os.environ.get("USE_JAX") == "1":
-        1
-    else:
-        import numba
-
-except ModuleNotFoundError:
-    logger.warning(
-        f"\n******************************************************************************\n"
-        f"Numba is not being used, either because it is disabled in `config/general.yaml` "
-        f"or because it is not installed.\n\n. "
-        f"This will lead to slow performance.\n\n. "
-        f"Install numba as described at the following webpage for improved performance. \n"
-        f"https://pyautolens.readthedocs.io/en/latest/installation/numba.html \n"
-        f"********************************************************************************"
-    )
-
 
 def jit(nopython=nopython, cache=cache, parallel=parallel):
-    def wrapper(func):
-        try:
-            use_numba = conf.instance["general"]["numba"]["use_numba"]
-
-            if not use_numba:
-                return func
 
-        except KeyError:
-            pass
+    def wrapper(func):
 
         try:
+
             import numba
 
             return numba.jit(func, nopython=nopython, cache=cache, parallel=parallel)
 
         except ModuleNotFoundError:
+
             return func
 
     return wrapper