From 1d2e783ae8c5687cc65c0f9a7eb8e3e9fae846bb Mon Sep 17 00:00:00 2001
From: Sebastian Hoffmann <shoffmann.git@gmail.com>
Date: Mon, 6 Jan 2025 15:30:52 +0100
Subject: [PATCH] fix: don't print devices twice

---
 dmlcloud/core/callbacks.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/dmlcloud/core/callbacks.py b/dmlcloud/core/callbacks.py
index e764877..fdcea00 100644
--- a/dmlcloud/core/callbacks.py
+++ b/dmlcloud/core/callbacks.py
@@ -460,10 +460,6 @@ def pre_run(self, pipe):
 
         diagnostics = general_diagnostics()
 
-        diagnostics += '\n* DEVICES:\n'
-        devices = all_gather_object(str(pipe.device))
-        diagnostics += '\n'.join(f'    - [Rank {i}] {device}' for i, device in enumerate(devices))
-
         diagnostics += '\n* CONFIG:\n'
         diagnostics += '\n'.join(f'    {line}' for line in OmegaConf.to_yaml(pipe.config, resolve=True).splitlines())