[BUG] update distributed path

xrsrke · Oct 23, 2023 · ada59a8 · ada59a8
1 parent 0e5479c
commit ada59a8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pipegoose/distributed/__init__.py b/pipegoose/distributed/__init__.py
@@ -0,0 +1,2 @@
+from pipegoose.distributed.parallel_context import ParallelContext
+from pipegoose.distributed.parallel_mode import ParallelMode