fix test

mllam · Oct 31, 2024 · 047a656 · 047a656
1 parent 5b8faaa
commit 047a656
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_distributed.py b/tests/test_distributed.py
@@ -32,14 +32,14 @@ def distributed():
     ],
 )
 def test_run_distributed(args):
-    if distributed() and "--dask-distributed-local-core-fraction" in args:
+    if distributed():
+        call_wrapper(args)
+    elif not distributed() and "--dask-distributed-local-core-fraction" in args:
         pytest.raises(
             ZeroDivisionError,
             call_wrapper(args),
             match="division by zero",
         )
-    elif distributed():
-        call_wrapper(args)
     else:
         pytest.raises(
             ModuleNotFoundError,