xfu314's picture
Add phantom project with submodules and dependencies
96da58e
"""
Numba utils.
"""
import numba
import robosuite.macros as macros
def jit_decorator(func):
if macros.ENABLE_NUMBA:
return numba.jit(nopython=True, cache=macros.CACHE_NUMBA)(func)
return func