I need to njit compile this function (with parallelization). But it throws the following error in the linewhere np.isnan() gets called. np.isnan() is supported, therefore: W