code: def conditional_jit(function=None, **kwargs):  # noqa: D202
      
  
      def wrapper(function):
          try:
              numba = importlib.import_module("numba")
              return numba.jit(**kwargs)(function)
  
          except ImportError:
              return function
  
      if function:
          return wrapper(function)
      else:
          return wrapper
docs: Use numba's jit decorator if numba is installed.
  
      Notes
      -----
          If called without arguments  then return wrapped function.
  
          @conditional_jit
          def my_func():
              return
  
          else called with arguments
  
          @conditional_jit(nopython=True)
          def my_func():
              return