diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py index 0b3ec54e..5b62406d 100644 --- a/dask_cuda/tests/test_spill.py +++ b/dask_cuda/tests/test_spill.py @@ -89,7 +89,7 @@ def delayed_worker_assert(total_size, device_chunk_overhead, serialized_chunk_ov "memory_limit": int(800e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, { @@ -97,7 +97,7 @@ def delayed_worker_assert(total_size, device_chunk_overhead, serialized_chunk_ov "memory_limit": int(200e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": True, }, { @@ -105,7 +105,7 @@ def delayed_worker_assert(total_size, device_chunk_overhead, serialized_chunk_ov "memory_limit": 0, "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, ], @@ -167,7 +167,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": int(800e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, { @@ -175,7 +175,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": int(200e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": True, }, { @@ -183,7 +183,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": 0, "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, ], @@ -243,7 +243,7 @@ async def test_cupy_cluster_device_spill(params): "memory_limit": int(800e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, { @@ -251,7 +251,7 @@ async def test_cupy_cluster_device_spill(params): "memory_limit": int(200e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": True, }, { @@ -259,7 +259,7 @@ async def test_cupy_cluster_device_spill(params): "memory_limit": 0, "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, ], @@ -332,7 +332,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": int(800e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, { @@ -340,7 +340,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": int(200e6), "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": True, }, { @@ -348,7 +348,7 @@ def test_device_spill(client, scheduler, worker): "memory_limit": 0, "host_target": 0.0, "host_spill": 0.0, - "host_pause": None, + "host_pause": 0.0, "spills_to_disk": False, }, ],