Corrected axes order for output of propagate_all

Author: Christopher-K-Long

src/qugrad/systems/_systems.py

@@ -556,7 +556,7 @@ def propagate_all(self, *args) -> np.ndarray[complex]:
 
         Returns
         -------
-        NDArray[Shape[n_time_steps+1, :attr:`state_shape`], complex]
+        NDArray[Shape[:attr:`state_shape`, n_time_steps+1], complex]
             The state at each integrator time step (including the initial
             state).
 

src/qugrad/systems/_systems.pyi

@@ -596,7 +596,7 @@ class QuantumSystem:
 
         Returns
         -------
-        NDArray[Shape[n_time_steps+1, :attr:`state_shape`], complex]
+        NDArray[Shape[:attr:`state_shape`, n_time_steps+1], complex]
             The state at each integrator time step (including the initial
             state).