rusticluftig commited on
Commit
7970a56
·
1 Parent(s): b47b4b4

Add custom endpoint

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import functools
2
  import traceback
3
  import gradio as gr
@@ -90,7 +91,12 @@ def run_with_retries(func, *args, **kwargs):
90
 
91
  def get_subtensor_and_metagraph() -> Tuple[bt.subtensor, bt.metagraph]:
92
  def _internal() -> Tuple[bt.subtensor, bt.metagraph]:
93
- subtensor = bt.subtensor("finney")
 
 
 
 
 
94
  metagraph = bt.metagraph(NETUID, lite=False)
95
  return subtensor, metagraph
96
 
 
1
+ import argparse
2
  import functools
3
  import traceback
4
  import gradio as gr
 
91
 
92
  def get_subtensor_and_metagraph() -> Tuple[bt.subtensor, bt.metagraph]:
93
  def _internal() -> Tuple[bt.subtensor, bt.metagraph]:
94
+ if SUBTENSOR_ENDPOINT:
95
+ parser = argparse.ArgumentParser()
96
+ bt.subtensor.add_args(parser)
97
+ subtensor = bt.subtensor(config=bt.config(parser=parser, args=["--subtensor.chain_endpoint", SUBTENSOR_ENDPOINT]))
98
+ else:
99
+ subtensor = bt.subtensor("finney")
100
  metagraph = bt.metagraph(NETUID, lite=False)
101
  return subtensor, metagraph
102