This script: import numpy as np import numba as nb from math import pi, atan2, sqrt @nb.njit(fastmath=True, locals=dict(w=nb.uint32)) def test2(im, grads, w): for i in range(im.size-w-1): dx = im[i+w+1]-im[i] dy = im[i+w]-im[i+1] grads[i] = (sqrt(dx*dx+dy*dy), np.int8(np.int8((atan2(…

I think you should be able to write it like: grads["magn"][i] = sqrt(dx*dx+dy*dy) grads["dir"][i] = np.int8(np.int8((atan2(dy, dx)+1.125*pi)/(pi/4)+5)%8)

[image] pauljurczak: The script runs fine without Numba. I though that Tuple(float32, int8) is the proper type to set an element of grads. What is the problem here? Related post

In general, fancy-dtype stuff is a bit hit-and-miss with numba. It can usually be worked around one way or another.

I don't understand this compilation error

Numba Support: What is this error message?

nelson2005 January 30, 2023, 2:56pm 3

Topic		Replies	Views
Getitem error number with numba function Community Support	5	1229	September 15, 2023
Numba dictionary is not working Support: What is this error message?	1	1162	May 10, 2023
How do I solve this numba.core.errors.TypingError? Support: What is this error message?	1	903	October 14, 2023
Jitclass with tuple not working Numba	3	441	October 11, 2022
Cannot determine Numba type of <class 'type'> Support: What is this error message?	9	11953	December 22, 2021

I don't understand this compilation error

Related topics