Merge pull request #1621 from PrefectHQ/fix-dask

Update dask client interface to account for kwarg changes in 2.6.0
PrefectHQ · Oct 16, 2019 · 70ef07a · 70ef07a
2 parents fddfede + 6df5c35
commit 70ef07a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/prefect/engine/executors/dask.py b/src/prefect/engine/executors/dask.py
@@ -67,12 +67,12 @@ def start(self) -> Iterator[None]:
         Creates a `dask.distributed.Client` and yields it.
         """
         try:
-            self.kwargs.update(
-                silence_logs=logging.CRITICAL if not self.debug else logging.WARNING
-            )
-            with Client(
-                self.address, processes=self.local_processes, **self.kwargs
-            ) as client:
+            if self.address is None:
+                self.kwargs.update(
+                    silence_logs=logging.CRITICAL if not self.debug else logging.WARNING
+                )
+                self.kwargs.update(processes=self.local_processes)
+            with Client(self.address, **self.kwargs) as client:
                 self.client = client
                 self.is_started = True
                 yield self.client