Переглянути джерело

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