add diloco rank

PrimeIntellect-ai · Sep 30, 2024 · 1c098b2 · 1c098b2
1 parent e080641
commit 1c098b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/zeroband/utils/world_info.py b/src/zeroband/utils/world_info.py
@@ -29,7 +29,7 @@ def __repr__(self):
 
     @property
     def diloco_rank(self):
-        return self.rank // self.local_world_size
+        return self.global_rank
 
 
 def get_world_info() -> WorldInfo: