Use default_initializer in WorkerProcesses (#6534)

dask · Jun 10, 2022 · b119eb9 · b119eb9
1 parent 43ca938
commit b119eb9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/distributed/nanny.py b/distributed/nanny.py
@@ -809,12 +809,10 @@ def _run(
         try:
             os.environ.update(env)
             dask.config.set(config)
-            try:
-                from dask.multiprocessing import initialize_worker_process
-            except ImportError:  # old Dask version
-                pass
-            else:
-                initialize_worker_process()
+
+            from dask.multiprocessing import default_initializer
+
+            default_initializer()
 
             if silence_logs:
                 logger.setLevel(silence_logs)