From 91268dbfaa8c0ddb1e1750255cb1b982e99b414a Mon Sep 17 00:00:00 2001
From: Mickus Timothee <mickus@lx7-humtdk75.hum.helsinki.fi>
Date: Tue, 19 Sep 2023 23:26:21 +0300
Subject: [PATCH] finally bothered to test it

---
 mammoth/distributed/__init__.py | 3 ---
 1 file changed, 3 deletions(-)

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",