From f4fd7d44d6aeb3c13271bb9e4f2144d03dc3b580 Mon Sep 17 00:00:00 2001
From: zoj <44142765+zoj613@users.noreply.github.com>
Date: Mon, 21 Dec 2020 00:11:34 +0200
Subject: [PATCH] FIX: correct input dimensionality check on the python side

---
 pyhtnorm/_htnorm.pyx | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/pyhtnorm/_htnorm.pyx b/pyhtnorm/_htnorm.pyx
index a0c067b..e20ecee 100644
--- a/pyhtnorm/_htnorm.pyx
+++ b/pyhtnorm/_htnorm.pyx
@@ -57,9 +57,7 @@ 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
-    -------
+Methods -------
     hyperplane_truncated_mvnorm(mean, cov, g, r, diag=False, out=False)
     structured_precision_mvnorm(mean, a, phi, omega, mean_structured=False,
                                 a_type=0, o_type=0, out=None)
@@ -257,7 +255,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"
             )