From e2311030d117fd4185eac5c94cfae43a14bdc3bb Mon Sep 17 00:00:00 2001
From: Hauke Schulz <43613877+observingClouds@users.noreply.github.com>
Date: Thu, 31 Oct 2024 11:03:09 +0100
Subject: [PATCH] fix test

---
 tests/test_distributed.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/test_distributed.py b/tests/test_distributed.py
index 40eba09..659680d 100644
--- 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",
         )