소스 검색

make run_server executable via python -m

justheuristic 5 년 전
부모
커밋
71a28cc27f
1개의 변경된 파일8개의 추가작업 그리고 1개의 파일을 삭제
  1. 8 1
      tests/test_utils/run_server.py

+ 8 - 1
tests/test_utils/run_server.py

@@ -22,10 +22,17 @@ def make_dummy_server(host='0.0.0.0', port=None, num_experts=1, expert_cls='ffn'
     # initialize network
     network = None
     if not no_network:
+        if not len(initial_peers):
+            print("No initial peers provided. Starting additional network as an initial peer.")
+            network = tesseract.TesseractNetwork(
+                *initial_peers, port=network_port or tesseract.find_open_port(), start=True)
+            print(f"Running DHT root on port {network.port}")
+        else:
+            print("Bootstrapping dht with peers:", initial_peers)
+
         network = tesseract.TesseractNetwork(
             *initial_peers, port=network_port or tesseract.find_open_port(), start=True)
         if verbose:
-            print("Parsed initial peers:", initial_peers)
             print(f"Running network node on port {network.port}")
 
     # initialize experts