next try for jit option with nopython

sibirrer · Jul 11, 2023 · e7c12c9 · e7c12c9
1 parent 04eb495
commit e7c12c9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lenstronomy/Util/image_util.py b/lenstronomy/Util/image_util.py
@@ -130,6 +130,8 @@ def re_size_array(x_in, y_in, input_values, x_out, y_out):
     :param y_out:
     :return:
     """
+    # from skimage.transform import resize
+    # resize(input_values)
     interp_2d = interpolate.interp2d(x_in, y_in, input_values, kind='linear')
     # interp_2d = scipy.interpolate.RectBivariateSpline(x_in, y_in, input_values, kx=1, ky=1)
     out_values = interp_2d.__call__(x_out, y_out)

diff --git a/lenstronomy/Util/numba_util.py b/lenstronomy/Util/numba_util.py
@@ -51,8 +51,8 @@ def overload(nopython=nopython, cache=cache, parallel=parallel, fastmath=fastmat
         def wrapper(func):
             return func
 
-        @numba.extending.overload(wrapper, nopython=nopython, cache=cache, parallel=parallel, fastmath=fastmath,
-                                  error_model=error_model)
+        @numba.extending.overload(wrapper, jit_options={'nopython': nopython, 'cache': cache, 'parallel': parallel,
+                                                        'fastmath': fastmath, 'error_model': error_model})
         def ol_wrapper(func):
             return func
     else: