fix test

mllam · Oct 31, 2024 · e231103 · e231103
1 parent 7e49f90
commit e231103
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/test_distributed.py b/tests/test_distributed.py
@@ -40,18 +40,21 @@ def test_run_distributed(args):
         if core_fraction > 0:
             pytest.raises(
                 ZeroDivisionError,
-                call_wrapper(args),
+                call_wrapper,
+                args=(args,),
                 match="division by zero",
             )
         else:
             pytest.raises(
                 ModuleNotFoundError,
-                call_wrapper(args),
+                call_wrapper,
+                args=(args,),
                 match="Currently dask.distrubuted is not installed",
             )
     else:
         pytest.raises(
             ModuleNotFoundError,
-            call_wrapper(args),
+            call_wrapper,
+            args=(args,),
             match="Currently dask.distrubuted is not installed",
         )