ModuleNotFoundError: No module named 'sklearn.externals.six'

[dhdistc]

ModuleNotFoundError Traceback (most recent call last)
in
2 import matplotlib.pyplot as plt
3 import pandas as pd
----> 4 import mglearn

e:\tools\python\lib\site-packages\mglearn_init_.py in
----> 1 from . import plots
2 from . import tools
3 from .plots import cm3, cm2
4 from .tools import discrete_scatter
5 from .plot_helpers import ReBl

e:\tools\python\lib\site-packages\mglearn\plots.py in
1 from .plot_linear_svc_regularization import plot_linear_svc_regularization
----> 2 from .plot_interactive_tree import plot_tree_progressive, plot_tree_partition
3 from .plot_animal_tree import plot_animal_tree
4 from .plot_rbf_svm_parameters import plot_svm
5 from .plot_knn_regression import plot_knn_regression

e:\tools\python\lib\site-packages\mglearn\plot_interactive_tree.py in
4 from sklearn.tree import DecisionTreeClassifier
5
----> 6 from sklearn.externals.six import StringIO # doctest: +SKIP
7 from sklearn.tree import export_graphviz
8 from imageio import imread

ModuleNotFoundError: No module named 'sklearn.externals.six'

[amueller]

Thanks for reporting. You can change the import to from io import StringIO locally if you like, I'll push a fix shortly.

[amueller]

pushed a fix to this repo and mglearn, let me know if it works!