snsd0805/gpu-provider
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: some ui commands

Browse Source
This commit is contained in:
snsd0805 2024-05-31 22:25:29 +08:00
parent 6093742092
commit b24d96ba7c
Signed by: snsd0805
GPG Key ID: 569349933C77A854
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/node_manager.py
View File

@ -85,7 +85,7 @@ class NodeManager():
print(f"And we have load your wallet private key {WALLET_KEY} (address={self.wallet.address})") print(f"And we have load your wallet private key {WALLET_KEY} (address={self.wallet.address})")
print() print()
if self.w3.is_connected(): if self.w3.is_connected():
print("Connected Successfully.") print("[INFO] Connected Successfully.")
print() print()
# Register the cluster # Register the cluster
@ -94,7 +94,7 @@ class NodeManager():
print(f"\nWe will register this cluster({self.GPU} * {gpu_num})...") print(f"\nWe will register this cluster({self.GPU} * {gpu_num})...")
receipt = self.scheduler.register_cluster(self.wallet, gpu_id, gpu_num) receipt = self.scheduler.register_cluster(self.wallet, gpu_id, gpu_num)
event = self.scheduler.get_cluster_event(receipt) event = self.scheduler.get_cluster_event(receipt)
print("\nOK. this is our cluster event on the blockchain") print("\n[INFO] Register our cluster succefully. \nThis is our cluster event on the blockchain: ")
print(f" {event[0]['args']}") print(f" {event[0]['args']}")
# start waiting # start waiting
@ -103,9 +103,22 @@ class NodeManager():
print("Ctrl+C to stop the waiting...") print("Ctrl+C to stop the waiting...")
try: try:
next_task = self.scheduler.listen_task(self.wallet.address) next_task = self.scheduler.listen_task(self.wallet.address)
print(next_task)
except: except:
print("[STOP] stop the waiting") print("[INFO] stop the waiting")
return
# get task info
task_index = next_task['args']['taskIndex']
data_image = next_task['args']['dataImage']
train_image = next_task['args']['trainImage']
print("\n[INFO] You Receive a new task:")
print(f" - Download Image: {data_image}")
print(f" - Training Image: {train_image}")
# Start Downloading
else: else:
print("[ERROR] Connected Failed.") print("[ERROR] Connected Failed.")