From ee8e2b6f515a4d703218d3ea7cedcb31ba5b0809 Mon Sep 17 00:00:00 2001 From: Thierry Moudiki Date: Sat, 27 Apr 2024 11:34:36 +0200 Subject: [PATCH] fix ENet regressor Pt.4 --- mlsauce/elasticnet/__init__.py | 5 ++++- setup.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/mlsauce/elasticnet/__init__.py b/mlsauce/elasticnet/__init__.py index 2cbe47c..b73aa88 100644 --- a/mlsauce/elasticnet/__init__.py +++ b/mlsauce/elasticnet/__init__.py @@ -1,3 +1,6 @@ -from .enet import ElasticNetRegressor +try: + from .enet import ElasticNetRegressor +except ImportError: + pass __all__ = ["ElasticNetRegressor"] \ No newline at end of file diff --git a/setup.py b/setup.py index d56e1b9..eeb744e 100644 --- a/setup.py +++ b/setup.py @@ -177,6 +177,7 @@ def setup_package(): "numpy", "Cython", "joblib", + "matplotlib", "pandas", "requests", "scikit-learn",