\usepackage{algorithm, algcompatible, algpseudocode}
\begin{algorithm}\caption{Steps of computation in a full GN block}\label{alg:gn_block}
\begin{algorithmic}[1]
\Function{GraphNetwork}{$E, V, u$}
\For{$k \in \{1 \dots N_e\}$}
\State $e_k' \gets \phi^e(e_k, E_{r_k}, E_{s_k}, u)$ \Comment{1. Compute updated edge attributes}
\EndFor
\For{$i \in \{1 \dots N_v\}$}
\State $E_i' \gets \{(e_k', r_k, s_k)\}_{r_k = i, k = 1:N_e}$
\State $\Bar{e}_i' \gets \rho^{e \to v}(E_i')$ \Comment{2. Aggregate edge attributes per node}
\State $E_i' \gets \phi^v(\Bar{e}_i', E_i, u)$ \Comment{3. Compute updated node attributes}
\EndFor
\State $E' \gets \{E_i'\}_{i = 1:N_v}$
\State $E' \gets \{(e_k', r_k, s_k)\}_{k = 1:N_e}$
\State $\Bar{e}' \gets \rho^{e \to u}(E')$ \Comment{4. Aggregate edge attributes globally}
\State $\Bar{v}' \gets \rho^{v \to u}(E')$ \Comment{5. Aggregate node attributes globally}
\State $u' \gets \phi^u(\Bar{e}', \Bar{v}', u)$ \Comment{6. Compute updated global attribute}
\State \Return $(E', V', u')$
\EndFunction
\end{algorithmic}
\end{algorithm}
This is the code I have.. I want the triangles to be exactly one below the other (like in the image), not that the comments are aligned to the right. 

