Update _stump_classifier.py

Techtonique · Apr 24, 2024 · 0ffb0b8 · 0ffb0b8
1 parent d875514
commit 0ffb0b8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mlsauce/stump/_stump_classifier.py b/mlsauce/stump/_stump_classifier.py
@@ -2,8 +2,11 @@
 from sklearn.base import BaseEstimator
 from sklearn.base import ClassifierMixin
 
-import pyximport; pyximport.install() 
-from . import _stumpc as stumpc
+try:
+    import pyximport; pyximport.install() 
+    from . import _stumpc as stumpc
+except:
+    pass
 
 
 class StumpClassifier(BaseEstimator, ClassifierMixin):