Skip to content

cellects.utils.decorators

cellects.utils.decorators

This module modifies the njit decorator to make it optional.

njit(*args, **kwargs)

numba n-jit decorator that can be disabled. Useful for testing.

Source code in src/cellects/utils/decorators.py 
10
11
12
13
14
15
16
17
18
def njit(*args, **kwargs):
    """ numba n-jit decorator that can be disabled. Useful for testing.
    """
    if USE_NUMBA:
        return _real_njit(*args, **kwargs)
    # test mode: return identity decorator
    def deco(func):
        return func
    return deco