Update memory_pause_fraction in test_spill (#757)

Changes from dask/distributed#5438 have slightly modified the behavior for worker spilling options, thus we need to update memory_pause_fraction to a value that isn't `None`. Authors: - Peter Andreas Entschev (https://github.com/pentschev) Approvers: - Benjamin Zaitlen (https://github.com/quasiben) URL: #757
rapidsai · Oct 20, 2021 · e9a291a · e9a291a
1 parent 05c95f0
commit e9a291a
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py
@@ -89,23 +89,23 @@ 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,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": int(200e6),
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": True,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": 0,
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": False,
         },
     ],
@@ -167,23 +167,23 @@ 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,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": int(200e6),
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": True,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": 0,
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": False,
         },
     ],
@@ -243,23 +243,23 @@ 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,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": int(200e6),
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": True,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": 0,
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": False,
         },
     ],
@@ -332,23 +332,23 @@ 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,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": int(200e6),
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": True,
         },
         {
             "device_memory_limit": int(200e6),
             "memory_limit": 0,
             "host_target": 0.0,
             "host_spill": 0.0,
-            "host_pause": None,
+            "host_pause": 0.0,
             "spills_to_disk": False,
         },
     ],