yes I will update the PR. I was thinking of trying to to put the whole method as `nogil` and wrap the leading numpy calls with `with gil` block.