Skip to content

Commit

Permalink
smaller
Browse files Browse the repository at this point in the history
  • Loading branch information
TomAugspurger committed Aug 29, 2018
1 parent d9b4d9a commit 4342853
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions tests/test_dask_based.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@


class TestDaskMatches(unittest.TestCase):
maxDiff = None

def test_dask_matches(self):
with dask.config.set(scheduler='single-threaded'):
Expand All @@ -44,9 +43,9 @@ def test_dask_matches(self):

self.assertEqual(a.score(X, y), b.score(X, y))
self.assertEqual(a.pareto_front_fitted_pipelines_.keys(),
b.pareto_front_fitted_pipelines_.keys())
b.pareto_front_fitted_pipelines_.keys())
self.assertEqual(a.evaluated_individuals_,
b.evaluated_individuals_)
b.evaluated_individuals_)

def test_handles_errors(self):
X, y = make_classification(n_samples=5)
Expand Down

0 comments on commit 4342853

Please sign in to comment.