瀏覽代碼

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