fix import in discretize

marcotcr · Sep 21, 2017 · 1920bba · 1920bba
1 parent 95a948b
commit 1920bba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lime/discretize.py b/lime/discretize.py
@@ -3,6 +3,7 @@
 """
 import numpy as np
 import sklearn
+import sklearn.tree
 from abc import ABCMeta, abstractmethod
 
 

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='lime',
-      version='0.1.1.23',
+      version='0.1.1.24',
       description='Local Interpretable Model-Agnostic Explanations for machine learning classifiers',
       url='http://github.com/marcotcr/lime',
       author='Marco Tulio Ribeiro',