Merge pull request #82 from smartnodes-lab/testnet-v2

Minor fix for worker config and secondary script switching.
Added worker bootstrap to __init__

Author: mattjhawken

bin/config.json

@@ -1,7 +1,7 @@
 {
   "address": "0x1Bc3a15dfFa205AA24F6386D959334ac1BF27336",
-  "local": "false",
-  "trusted": "false",
-  "mining": "false",
-  "mining-script": "path/to/mining.executable",
+  "local": false,
+  "trusted": false,
+  "mining": false,
+  "mining-script": "path/to/mining.executable"
 }

bin/run_worker.py

@@ -57,8 +57,19 @@ def start_mining(mining_script, use_sudo=False):
     """
     Start the mining process using the specified script.
     """
-    command = f"sudo {mining_script}" if use_sudo else mining_script
-    return subprocess.Popen(command, shell=True)
+    # Convert to absolute path relative to current working directory
+    if not os.path.isabs(mining_script):
+        mining_script = os.path.abspath(mining_script)
+
+    # Verify script exists
+    if not os.path.exists(mining_script):
+        raise FileNotFoundError(f"Mining script not found: {mining_script}")
+
+    if use_sudo:
+        # Use list form for better security
+        return subprocess.Popen(["sudo", mining_script])
+    else:
+        return subprocess.Popen([mining_script])
 
 
 def stop_mining(mining_process):

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tensorlink"
-version = "0.2.0"
+version = "0.2.0.post1"
 description = "Tensorlink is a decentralized Python library for distributed PyTorch models, providing easy Hugging Face API access and enabling users to share compute resources across a global network."
 authors = [
     "Smartnodes Lab <[email protected]>",

tensorlink/nodes/worker.py

@@ -71,27 +71,28 @@ def __init__(
 
             self.dht.store(hashlib.sha256(b"ADDRESS").hexdigest(), self.public_key)
 
-            # if self.local_test is False:
-            #     attempts = 0
-            #
-            #     self.debug_print("Bootstrapping...", tag="Worker")
-            #     while attempts < 3 and len(self.validators) == 0:
-            #         self.bootstrap()
-            #         if len(self.validators) == 0:
-            #             time.sleep(15)
-            #             self.debug_print(
-            #                 "No validators found, trying again...", tag="Worker"
-            #             )
-            #             attempts += 1
-            #
-            #     if len(self.validators) == 0:
-            #         self.debug_print(
-            #             "No validators found, shutting down...",
-            #             level=logging.CRITICAL,
-            #             tag="Worker",
-            #         )
-            #         self.stop()
-            #         self.terminate_flag.set()
+            if not self.local_test and not self.off_chain_test:
+                attempts = 0
+
+                self.debug_print("Bootstrapping...", tag="Worker")
+                while attempts < 3 and len(self.validators) == 0:
+                    self.bootstrap()
+                    if len(self.validators) == 0:
+                        time.sleep(15)
+                        self.debug_print(
+                            "No validators found, trying again...", tag="Worker"
+                        )
+                        attempts += 1
+
+                if len(self.validators) == 0:
+                    self.debug_print(
+                        "No validators found, shutting down...",
+                        level=logging.CRITICAL,
+                        tag="Worker",
+                    )
+                    self.stop()
+                    self.terminate_flag.set()
+
         self.keeper.load_previous_state()
 
     def handle_data(self, data: bytes, node: Connection):