Neglected to backpropagate this debug output fix from my training branch

f02e62c9c0802332488ff52670691187c9d23f2f · 2024-11-12 06:31:26 -0800 · Nel Nibcord

Files touched

M exo/orchestration/standard_node.py

Diff

commit f02e62c9c0802332488ff52670691187c9d23f2f
Author: Nel Nibcord <blindcrone@tuta.io>
Date:   Tue Nov 12 06:31:26 2024 -0800

    Neglected to backpropagate this debug output fix from my training branch
---
 exo/orchestration/standard_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/orchestration/standard_node.py b/exo/orchestration/standard_node.py
index bd292976..c25e50d9 100644
--- a/exo/orchestration/standard_node.py
+++ b/exo/orchestration/standard_node.py
@@ -282,7 +282,7 @@ class StandardNode(Node):
       return
 
     next_partition_index = self.get_partition_index(offset = 1)
-    if DEBUG >= 1: print(f"Current partition index: {current_partition_index}")
+    if DEBUG >= 1: print(f"Next partition index: {next_partition_index}")
     if next_partition_index is not None:
       target_id = self.partitioning_strategy.partition(self.topology)[next_partition_index].node_id
       next_shard = self.get_current_shard(base_shard, next_partition_index)

← 03924cf9 Need tokens. Also, for some reason this gets mad if we have · back to Exo · fix debug log 19463737 →