FIX: correct input dimensionality check on the python side

zoj613 · Dec 21, 2020 · 0719999 · 0719999
1 parent 6986df7
commit 0719999
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyhtnorm/_htnorm.pyx b/pyhtnorm/_htnorm.pyx
@@ -57,7 +57,6 @@ cdef class HTNGenerator:
         The name of the generator to use for random number generation. The
         value needs to be one of {'pcg', 'xrs'}, where 'pcg' is PCG64 and 'xrs'
         is the Xoroshiro128plus bit generator.
-
     Methods
     -------
     hyperplane_truncated_mvnorm(mean, cov, g, r, diag=False, out=False)
@@ -257,7 +256,7 @@ cdef class HTNGenerator:
         if (omega.shape[0] != omega.shape[1]) or (a.shape[0] != a.shape[1]):
             raise ValueError('`omega` and `a` both need to be square matrices')
 
-        if (phi.shape[1] != omega.shape[0]) or (phi.shape[0] != a.shape[1]):
+        if (phi.shape[0] != omega.shape[0]) or (phi.shape[1] != a.shape[0]):
             raise ValueError(
                 "Shapes of `phi`, `omega` and `a` are not consistent"
             )