added tag to broadcast

aidecentralized · Sep 14, 2024 · 5ec2484 · 5ec2484
1 parent bfc6048
commit 5ec2484
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/communication/comm_utils.py b/src/utils/communication/comm_utils.py
@@ -60,7 +60,7 @@ def receive(self, node_ids: str|int|List[str|int], tag:int=0) -> Any:
         else:
             return self.comm.receive(node_ids)
 
-    def broadcast(self, data: Any):
+    def broadcast(self, data: Any, tag:int=0):
         self.comm.broadcast(data)
 
     def all_gather(self, tag:int=0):