Get an error when using np.take_along_axis function

Question

I am implementing an algorithm. Somewhere in the work I need to sort an array and use the sorted indexes. To do this I use the np.argsort function and to use the indexes ret