Nest dict with jit class

Numba Community Support
1

I am trying to use the jitclass function where one of my attributes is a nested dictionary.
Here is my code.

@jitclass([(‘dt’, types.int32),
(‘last_index_tick’, types.DictType(types.int32, types.DictType(types.unicode_type, types.int64)))])
class TickFeature:
def init(self, dt: int):
self.dt = dt
#self.last_index_tick = defaultdict(dict)
self.last_index_tick = numba.typed.Dict.empty(
key_type=types.int32,
value_type=types.DictType(types.unicode_type, types.int64)
)
self.init_index_tick()

def init_index_tick(self, ):
    l = np.array([905, 300, 16, 399005, 399006], dtype=np.int64)

    tmp = numba.typed.Dict.empty(key_type=types.unicode_type, value_type=types.int64)
    '''
    self.last_index_tick = numba.typed.Dict.empty(
        key_type=types.types.int32,
        value_type=types.DictType(types.unicode_type, types.int64)
    )
    '''
    for item in l:
        self.last_index_tick[item] = tmp

Here is the error message.

Failed in nopython mode pipeline (step: nopython frontend)
Failed in nopython mode pipeline (step: nopython frontend)
No implementation of function Function(<class ‘numba.core.types.containers.DictType’>) found for signature:

DictType(typeref[unicode_type], class(int64))

There are 2 candidate implementations:
- Of which 2 did not match due to:
Overload in function ‘DictType’: File: numba/core/extending.py: Line 37.
With argument(s): ‘(typeref[unicode_type], class(int64))’:
Rejected as the implementation raised a specific error:
TypeError: typer() takes 0 positional arguments but 2 were given
raised from /home/wenrui29/.opt/anaconda/lib/python3.7/site-packages/numba/core/typing/templates.py:363

During: resolving callee type: Function(<class ‘numba.core.types.containers.DictType’>)
During: typing of call at (9)

File “”, line 9:
def init(self, dt: int):

key_type=types.int32,
value_type=types.DictType(types.unicode_type, types.int64)
^

During: resolving callee type: jitclass.TickFeature#7fb898862590dt:int32,last_index_tick:DictType[int32,DictType[unicode_type,int64]]
During: typing of call at (3)

During: resolving callee type: jitclass.TickFeature#7fb898862590dt:int32,last_index_tick:DictType[int32,DictType[unicode_type,int64]]
During: typing of call at (3)

File “”, line 3:

2

In case anyone else has this issue, it is answered here: https://github.com/numba/numba/issues/6191#issuecomment-684022879