From 51dcff6f1c2ca9fd96f13063c8d33b88f0fb3b65 Mon Sep 17 00:00:00 2001
From: Thierry Moudiki <thierry.moudiki@gmail.com>
Date: Sat, 27 Apr 2024 11:10:39 +0200
Subject: [PATCH] fix ENet regressor

---
 mlsauce/__init__.py           | 4 +++-
 mlsauce/booster/_boosterc.pyx | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/mlsauce/__init__.py b/mlsauce/__init__.py
index 350d908..4cabad2 100644
--- a/mlsauce/__init__.py
+++ b/mlsauce/__init__.py
@@ -23,7 +23,7 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = "0.10.0"
+#__version__ = "0.10.0"
 
 
 # On OSX, we can get a runtime error due to multiple OpenMP libraries loaded
@@ -58,6 +58,7 @@
     from .adaopt import AdaOpt
     from .booster import LSBoostClassifier, LSBoostRegressor
     from .datasets import download
+    from .elasticnet import ElasticNetRegressor
     from .lasso import LassoRegressor
     from .ridge import RidgeRegressor
     from .stump import StumpClassifier
@@ -68,6 +69,7 @@
         "AdaOpt",
         "LSBoostClassifier",
         "StumpClassifier",
+        "ElasticNetRegressor",
         "LassoRegressor",
         "LSBoostRegressor",
         "RidgeRegressor",
diff --git a/mlsauce/booster/_boosterc.pyx b/mlsauce/booster/_boosterc.pyx
index 4b53f57..1af10aa 100644
--- a/mlsauce/booster/_boosterc.pyx
+++ b/mlsauce/booster/_boosterc.pyx
@@ -306,7 +306,7 @@ def fit_booster_regressor(double[:,::1] X, double[:] y,
   elif solver == "lasso": 
     fit_obj = LassoRegressor(reg_lambda = reg_lambda, backend = backend)
   else: 
-    fit_obj = LassoRegressor(reg_lambda = reg_lambda, alpha = alpha, backend = backend)  
+    fit_obj = ElasticNetRegressor(reg_lambda = reg_lambda, alpha = alpha, backend = backend)  
 
 
   for iter in iterator: