diff --git a/contact_map/tests/test_dask_runner.py b/contact_map/tests/test_dask_runner.py index d72cb18..f938a66 100644 --- a/contact_map/tests/test_dask_runner.py +++ b/contact_map/tests/test_dask_runner.py @@ -11,8 +11,10 @@ def test_dask_integration(self): # this is an integration test to check that dask works dask = pytest.importorskip('dask') # pylint: disable=W0612 distributed = pytest.importorskip('dask.distributed') - - client = distributed.Client() + # Explicitly set only 4 workers on Travis instead of 31 + # Fix copied from https://github.com/spencerahill/aospy/pull/220/files + cluster = distributed.LocalCluster(n_workers=4) + client = distributed.Client(cluster) filename = find_testfile("trajectory.pdb") dask_freq = DaskContactFrequency(client, filename, cutoff=0.075,