diff --git a/lime/tests/test_lime_tabular.py b/lime/tests/test_lime_tabular.py index 5e6bd33f..e4b0f07b 100644 --- a/lime/tests/test_lime_tabular.py +++ b/lime/tests/test_lime_tabular.py @@ -1,10 +1,10 @@ import unittest import numpy as np -import sklearn +import sklearn # noqa import sklearn.datasets import sklearn.ensemble -import sklearn.linear_model +import sklearn.linear_model # noqa from sklearn.datasets import load_iris, make_classification from sklearn.ensemble import RandomForestClassifier from sklearn.linear_model import Lasso diff --git a/lime/tests/test_lime_text.py b/lime/tests/test_lime_text.py index db8c382f..f029f0ca 100644 --- a/lime/tests/test_lime_text.py +++ b/lime/tests/test_lime_text.py @@ -1,6 +1,6 @@ import unittest -import sklearn +import sklearn # noqa from sklearn.datasets import fetch_20newsgroups from sklearn.feature_extraction.text import TfidfVectorizer from sklearn.linear_model import Lasso