From 1b970b5d0994b0879ef3597f14138cbffc3e8f82 Mon Sep 17 00:00:00 2001
From: RYCKEBOER Thomas <37658138+thomasryck@users.noreply.github.com>
Date: Mon, 30 Sep 2024 13:17:49 +0200
Subject: [PATCH] debug test

---
 cyanure/estimators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cyanure/estimators.py b/cyanure/estimators.py
index c4a7359e3..3dd7038fa 100644
--- a/cyanure/estimators.py
+++ b/cyanure/estimators.py
@@ -287,7 +287,7 @@ def fit(self, X, labels, le_parameter=None):
 
         if (self.multi_class == "multinomial" or
            (self.multi_class == "auto" and not self._binary_problem)) and self.loss == "logistic":
-            if len(np.unique(labels)) != 2:
+            if len(np.unique(labels)) != 2 or self.multi_class == "multinomial":
                 self._binary_problem = False
 
                 loss = "multiclass-logistic"