fix ENet regressor Pt.4

Techtonique · Apr 27, 2024 · ee8e2b6 · ee8e2b6
1 parent 7ab5328
commit ee8e2b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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"]
diff --git a/setup.py b/setup.py
@@ -177,6 +177,7 @@ def setup_package():
         "numpy",
         "Cython",
         "joblib",
+        "matplotlib",
         "pandas",
         "requests",
         "scikit-learn",