Update _lasso.py

Techtonique · Apr 24, 2024 · 97f8fed · 97f8fed
1 parent d17176f
commit 97f8fed
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mlsauce/lasso/_lasso.py b/mlsauce/lasso/_lasso.py
@@ -6,8 +6,11 @@
 from sklearn.base import RegressorMixin
 from numpy.linalg import inv
 
-import pyximport; pyximport.install() 
-from . import _lassoc as mo
+try:
+    import pyximport; pyximport.install() 
+    from . import _lassoc as mo
+except:
+    pass 
 
 from ..utils import get_beta