typos

Helsinki-NLP · Sep 19, 2023 · 09b36aa · 09b36aa
1 parent 91268db
commit 09b36aa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mammoth/distributed/__init__.py b/mammoth/distributed/__init__.py
@@ -2,6 +2,7 @@
 from mammoth.distributed.communication import (
     all_gather_list,
     batch_producer,
+    consumer,
     broadcast_tensors,
     only_ready_reduce_and_rescale_grads,
     ErrorHandler,
@@ -18,6 +19,7 @@
     "all_gather_list",
     "batch_producer",
     "broadcast_tensors",
+    "consumer",
     "only_ready_reduce_and_rescale_grads",
     "ErrorHandler",
     "DeviceContext",