Add init imports

dask · Jan 31, 2023 · c04c580 · c04c580
1 parent 54d3da8
commit c04c580
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/distributed/shuffle/__init__.py b/distributed/shuffle/__init__.py
@@ -1,10 +1,13 @@
 from __future__ import annotations
 
+from distributed.shuffle._merge import HashJoinP2PLayer, hash_join_p2p
 from distributed.shuffle._scheduler_extension import ShuffleSchedulerExtension
 from distributed.shuffle._shuffle import P2PShuffleLayer, rearrange_by_column_p2p
 from distributed.shuffle._worker_extension import ShuffleWorkerExtension
 
 __all__ = [
+    "hash_join_p2p",
+    "HashJoinP2PLayer",
     "P2PShuffleLayer",
     "rearrange_by_column_p2p",
     "ShuffleSchedulerExtension",