Update _lasso.py - - use pyximport

Techtonique · Apr 23, 2024 · 13f224e · 13f224e
1 parent 83c27eb
commit 13f224e
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/mlsauce/lasso/_lasso.py b/mlsauce/lasso/_lasso.py
@@ -5,11 +5,10 @@
 from sklearn.base import BaseEstimator
 from sklearn.base import RegressorMixin
 from numpy.linalg import inv
-try:
-    from . import _lassoc as mo
-except ImportError:
-    import pyximport; pyximport.install() 
-    import _lassoc as mo
+
+import pyximport; pyximport.install() 
+import _lassoc as mo
+
 from ..utils import get_beta
 
 if platform.system() in ("Linux", "Darwin"):