Update devcontainer configuration and scripts

nanlabs · Feb 27, 2024 · 40d4b6f · 40d4b6f
1 parent 8946557
commit 40d4b6f
Show file tree

Hide file tree

Showing 7 changed files with 25 additions and 6 deletions.
diff --git a/.dependencies_installed b/.dependencies_installed
diff --git a/.devcontainer/compose.yml b/.devcontainer/compose.yml
@@ -23,4 +23,7 @@ services:
 
     # Overrides default command so things don't shut down after the process ends.
     command: /bin/sh -c "while sleep 1000; do :; done"
+
+    environment:
+      - CARGO_HOME=/home/devcontainer/.cargo
 
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -18,6 +18,10 @@
 	// connected. This is typically a file mount in .devcontainer/compose.yml
 	"workspaceFolder": "/workspaces/${localWorkspaceFolderBasename}",
 	"features": {
+		"ghcr.io/devcontainers/features/rust:1": {
+			"installTools": true,
+			"version": "latest"
+		},
 		"ghcr.io/devcontainers/features/python:1": {
 			"installTools": true,
 			"version": "latest"

diff --git a/scripts/run_miner.sh b/scripts/run_miner.sh
diff --git a/scripts/run_openai_miner.sh b/scripts/run_openai_miner.sh
@@ -0,0 +1,10 @@
+#!/bin/bash
+
+NETUID=${1:-1}  
+CHAIN_ENDPOINT=${2:-"ws://127.0.0.1:9946"}  
+WALLET_NAME=${3:-"openai_miner"}  
+WALLET_HOTKEY=${4:-"default"}  
+
+python -m pip install --upgrade --user -r ./neurons/miners/openai/requirements.txt
+
+python ./neurons/miners/openai/miner.py --netuid $NETUID --subtensor.chain_endpoint $CHAIN_ENDPOINT --wallet.name $WALLET_NAME --wallet.hotkey $WALLET_HOTKEY --logging.debug 
diff --git a/scripts/run_validator.sh b/scripts/run_validator.sh
@@ -1,3 +1,8 @@
-#!/bin/bash
+#!/bin/bash  
 
-python neurons/validators/validator.py --netuid 1 --subtensor.chain_endpoint ws://127.0.0.1:9946 --wallet.name validator --wallet.hotkey default --logging.debug
+NETUID=${1:-1}  
+CHAIN_ENDPOINT=${2:-"ws://127.0.0.1:9946"}  
+WALLET_NAME=${3:-"validator"}  
+WALLET_HOTKEY=${4:-"default"}  
+
+python3 neurons/validators/validator.py --netuid $NETUID --subtensor.chain_endpoint $CHAIN_ENDPOINT --wallet.name $WALLET_NAME --wallet.hotkey $WALLET_HOTKEY --logging.debug 
diff --git a/scripts/setup_devcontainer.sh b/scripts/setup_devcontainer.sh
@@ -3,7 +3,7 @@
 # Logic for setting up and running the environment
 setup_environment() {
     # Install the bittensor-prompting-example python package
-    python -m pip install -e .
+    python -m pip install --upgrade --user -e .
 
     # Create and set up wallets
     # This section can be skipped if wallets are already set up