Fix numba depreceation notice (#1233)

From numba 0.57 raise a warning if `nopython` is not supplied:
https://numba.readthedocs.io/en/stable/reference/deprecation.html#deprecation-of-object-mode-fall-back-behaviour-when-using-jit
This commit is contained in:
Paul Willot 2023-05-05 08:48:06 +02:00 committed by GitHub
parent b1c0815c79
commit 7ca9fbea86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -54,7 +54,7 @@ def median_filter(x: torch.Tensor, filter_width: int):
return result
@numba.jit
@numba.jit(nopython=True)
def backtrace(trace: np.ndarray):
i = trace.shape[0] - 1
j = trace.shape[1] - 1