Sure. Will also add a test. Not sure about the failure, I think `gradient` possibly doesn't work with multidimensional datetime input, but I find it a bit puzzling that advanced indexing worked and normal doesn't. It might have something to do with `arr[[0]]` returning an array while `arr[0,]` returns a scalar...