def get_moran_drift(M, T): k = M.shape[1] if k == 2: return wrightcore.create_moran_drift_rate_k2(M, T) elif k == 3: return wrightcore.create_moran_drift_rate_k3(M, T) elif k == 4: return wrightcore.create_moran_drift_rate_k4(M, T) else: raise NotImplementedError
def get_moran_drift(M, T): #FIXME: this is dumb, #FIXME: but I'm not sure how to use variable-dimension ndarrays in cython k = M.shape[1] if k == 2: return wrightcore.create_moran_drift_rate_k2(M, T) elif k == 3: return wrightcore.create_moran_drift_rate_k3(M, T) elif k == 4: return wrightcore.create_moran_drift_rate_k4(M, T) else: raise NotImplementedError