diff --git a/mlsauce/stump/_stump_classifier.py b/mlsauce/stump/_stump_classifier.py index c8a151c..9dbea06 100644 --- 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):