Or just the function checks if you passed in an array or a trajectory, and extracts the array from the trajectory if necessary. I dunno. It just seems silly to have so many docstrings for so little code.