My mistake, I didn't read the first `if`

`elif self.sigma.ndim == 2:` on the higher if should be clearer to understand. ?