Great, do I need to change this too?

Also for the swap_column case it is enough to do this right?

    def swap_csr_column(X, m, n):
        return swap_csc_row(X.T, m, n)