finally bothered to test it

Helsinki-NLP · Sep 19, 2023 · 91268db · 91268db
1 parent 22d147f
commit 91268db
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/mammoth/distributed/__init__.py b/mammoth/distributed/__init__.py
@@ -1,10 +1,8 @@
 """Module defining distributed communications utilities."""
 from mammoth.distributed.communication import (
     all_gather_list,
-    all_gather_stats,
     batch_producer,
     broadcast_tensors,
-    consummer,
     only_ready_reduce_and_rescale_grads,
     ErrorHandler,
 )
@@ -20,7 +18,6 @@
     "all_gather_list",
     "batch_producer",
     "broadcast_tensors",
-    "consummer",
     "only_ready_reduce_and_rescale_grads",
     "ErrorHandler",
     "DeviceContext",