From f64c4407afec046ee935dca1ce890274b1ea538e Mon Sep 17 00:00:00 2001
From: Lucas Bulgarelli <lucas.bulgarelli@hotmail.com>
Date: Tue, 19 Nov 2019 14:41:18 -0500
Subject: [PATCH] better test for transposing

---
 pyroc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyroc.py b/pyroc.py
index 75dad7e..41f7bce 100644
--- a/pyroc.py
+++ b/pyroc.py
@@ -114,7 +114,7 @@ def _parse_inputs(self, preds, target):
                 parsed = np.array([preds])
 
             # If needed to transpose matrix
-            if parsed.shape[1] != len(target):
+            if parsed.shape[0] == len(target):
                 parsed = preds.T
 
             # Use column names if it is a DataFrame