Merge a417ef9 into 5d278db

DistrictDataLabs · Apr 21, 2018 · 95a9897 · 95a9897
2 parents 5d278db + a417ef9
commit 95a9897
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_contrib/test_statsmodels/test_base.py b/tests/test_contrib/test_statsmodels/test_base.py
@@ -21,7 +21,7 @@
 from yellowbrick.contrib.statsmodels import StatsModelsWrapper
 
 try:
-    import statsmodels as sm
+    import statsmodels.api as sm
 except ImportError:
     sm = None
 

diff --git a/yellowbrick/contrib/statsmodels/base.py b/yellowbrick/contrib/statsmodels/base.py
@@ -30,7 +30,7 @@ class StatsModelsWrapper(BaseEstimator):
     --------
     First import the external libraries and helper utilities:
 
-    >>> import statsmodels as sm
+    >>> import statsmodels.api as sm
     >>> from functools import partial
 
     Instantiate a partial with the statsmodels API: