Update _ridge.py - - use pyximport

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