fix test

mllam · Oct 31, 2024 · 20ff275 · 20ff275
1 parent 047a656
commit 20ff275
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_distributed.py b/tests/test_distributed.py
@@ -34,7 +34,7 @@ def distributed():
 def test_run_distributed(args):
     if distributed():
         call_wrapper(args)
-    elif not distributed() and "--dask-distributed-local-core-fraction" in args:
+    elif not distributed() and args["dask-distributed-local-core-fraction"] > 0:
         pytest.raises(
             ZeroDivisionError,
             call_wrapper(args),