Minimal change to work around #7726 / support of UCX

dask · Apr 5, 2023 · d62693d · d62693d
1 parent 9eb8bb3
commit d62693d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/distributed/__init__.py b/distributed/__init__.py
@@ -7,7 +7,14 @@
 # isort: on
 
 import atexit
+import weakref
 
+# This finalizer registers an atexit handler that has to happen before
+# distributed registers its handlers, otherwise we observe hangs on
+# cluster shutdown when using the UCX comms backend. See
+# https://github.com/dask/distributed/issues/7726 for more discussion
+
+weakref.finalize(lambda: None, lambda: None)
 import dask
 from dask.config import config  # type: ignore