diff --git a/mammoth/distributed/__init__.py b/mammoth/distributed/__init__.py index 3250b46d..3be56fdb 100644 --- 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",