initialize node id to uuid4 if not set

05b9fa497d5086289ef1855c052626216844e26c · 2024-07-14 21:46:30 -0700 · Alex Cheema

Files touched

M main.py

Diff

commit 05b9fa497d5086289ef1855c052626216844e26c
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sun Jul 14 21:46:30 2024 -0700

    initialize node id to uuid4 if not set
---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 5e337d4e..58363caa 100644
--- a/main.py
+++ b/main.py
@@ -3,6 +3,7 @@ import asyncio
 import signal
 import mlx.core as mx
 import mlx.nn as nn
+import uuid
 from typing import List
 from exo.orchestration.standard_node import StandardNode
 from exo.networking.grpc.grpc_server import GRPCServer
@@ -13,7 +14,7 @@ from exo.topology.ring_memory_weighted_partitioning_strategy import RingMemoryWe
 
 # parse args
 parser = argparse.ArgumentParser(description="Initialize GRPC Discovery")
-parser.add_argument("--node-id", type=str, default="node1", help="Node ID")
+parser.add_argument("--node-id", type=str, default=str(uuid.uuid4()), help="Node ID")
 parser.add_argument("--node-host", type=str, default="0.0.0.0", help="Node host")
 parser.add_argument("--node-port", type=int, default=8080, help="Node port")
 parser.add_argument("--listen-port", type=int, default=5678, help="Listening port for discovery")

← ff597d95 fix discovery · back to Exo · add uuid requirement 53a5b3fc →