Skip to content

Commit

Permalink
Update test utility function name
Browse files Browse the repository at this point in the history
  • Loading branch information
morganjwilliams committed Jan 24, 2023
1 parent d92b587 commit b5407d6
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions test/util/skl/util_skl_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

HAVE_SKLEARN = True

def test_classifier():
def get_classifier():
param_grid = dict(gamma=np.array([0.001, 0.01]), C=np.array([1, 10]))
gs = GridSearchCV(SVC(gamma="scale"), param_grid, cv=2)
return gs
Expand All @@ -27,7 +27,7 @@ def test_classifier():
class TestFitSaveClassifier(unittest.TestCase):
def setUp(self):
self.X, self.y = sklearn.datasets.load_iris(return_X_y=True)
self.classifier = test_classifier()
self.classifier = get_classifier()
self.dir = temp_path()

def test_default(self):
Expand All @@ -46,7 +46,7 @@ def tearDown(self):
class TestClassifierPerformanceReport(unittest.TestCase):
def setUp(self):
self.X, self.y = sklearn.datasets.load_iris(return_X_y=True)
self.classifier = test_classifier()
self.classifier = get_classifier()
self.dir = temp_path()

def test_default(self):
Expand Down

0 comments on commit b5407d6

Please sign in to comment.