Update _stump_classifier.py - - use pyximport

Techtonique · Apr 23, 2024 · 6eb838b · 6eb838b
1 parent 8125c56
commit 6eb838b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mlsauce/stump/_stump_classifier.py b/mlsauce/stump/_stump_classifier.py
@@ -1,7 +1,11 @@
 import numpy as np
 from sklearn.base import BaseEstimator
 from sklearn.base import ClassifierMixin
-from . import _stumpc as stumpc
+try:
+    from . import _stumpc as stumpc
+except ImportError:
+    import pyximport; pyximport.install() 
+    import _stumpc as stumpc
 
 
 class StumpClassifier(BaseEstimator, ClassifierMixin):