removing useless prints

cantinilab · Jan 16, 2025 · 76ceb58 · 76ceb58
1 parent 26c3937
commit 76ceb58
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/rank_0.txt b/rank_0.txt
@@ -0,0 +1 @@
+Rank 0 completed test epoch
diff --git a/rank_1.txt b/rank_1.txt
@@ -0,0 +1 @@
+Rank 1 completed test epoch
diff --git a/scprint/model/model.py b/scprint/model/model.py
@@ -1247,9 +1247,6 @@ def on_validation_epoch_end(self):
             if torch.distributed.is_initialized():
                 # Set a timeout that's longer than your test typically takes
                 # Write rank to file for debugging
-                rank = torch.distributed.get_rank()
-                with open(f"rank_{rank}.txt", "a") as f:
-                    f.write(f"Rank {rank} completed test epoch\n")
                 self.trainer.strategy.barrier()
 
     def test_step(self, *args, **kwargs):