Add prometheus extrinsic

bittensor/_prometheus/__init__.py

@@ -55,7 +55,6 @@ def __new__(
         network: str = None,
         chain_endpoint: str = None,
         subtensor: 'bittensor.subtensor' = None,
-        prompt: bool = False, 
     ):
         """ Instantiates a global prometheus DB which can be accessed by other processes.
             Each prometheus DB is designated by a port.
@@ -76,8 +75,6 @@ def __new__(
                     If subtensor is not set, uses this network flag to create the subtensor connection.
                 chain_endpoint (default=None, type=str)
                     Overrides the network argument if not set.
-                prompt (:obj:`bool`, `optional`):
-                    If true, the call waits for confirmation from the user before proceeding.
         """
         if config == None:
             config = prometheus.config()
@@ -98,15 +95,13 @@ def __new__(
             subtensor = subtensor,
             port = config.prometheus.port,
             level = config.prometheus.level,
-            prompt = prompt,
         )
 
-    def serve(cls, wallet, subtensor, netuid, port, level, prompt):
+    def serve(cls, wallet, subtensor, netuid, port, level):
         serve_success = subtensor.serve_prometheus(
             wallet = wallet,
             port = port,
             netuid = netuid,
-            prompt = prompt
         )
         if serve_success and (level != prometheus.level.OFF.name):
             try:
@@ -121,7 +116,6 @@ def serve(cls, wallet, subtensor, netuid, port, level, prompt):
         else:
             logger.success('Prometheus:'.ljust(20) + '<red>OFF</red>')
             raise RuntimeError('Failed to serve neuron.')
-            return False
 
     @classmethod
     def config(cls) -> 'bittensor.Config':

bittensor/_subtensor/extrinsics/prometheus.py

@@ -49,8 +49,6 @@ def prometheus_extrinsic(
         wait_for_finalization (bool):
             if set, waits for the extrinsic to be finalized on the chain before returning true,
             or returns false if the extrinsic fails to be finalized within the timeout.
-        prompt (bool):
-            If true, the call waits for confirmation from the user before proceeding.
     Returns:
         success (bool):
             flag is true if extrinsic was finalized or uncluded in the block. 

bittensor/_subtensor/subtensor_impl.py

@@ -259,9 +259,8 @@ def serve_prometheus (
         netuid: int,
         wait_for_inclusion: bool = False,
         wait_for_finalization: bool = True,
-        prompt: bool = False,
     ) -> bool:
-        return prometheus_extrinsic( self, wallet = wallet, port = port, netuid = netuid, wait_for_inclusion = wait_for_inclusion, wait_for_finalization = wait_for_finalization, prompt = prompt)
+        return prometheus_extrinsic( self, wallet = wallet, port = port, netuid = netuid, wait_for_inclusion = wait_for_inclusion, wait_for_finalization = wait_for_finalization)
     #################
     #### Staking ####
     #################

tests/integration_tests/test_prometheus.py

@@ -42,5 +42,6 @@ def test_init_prometheus_success(self):
 
     def test_init_prometheus_failed(self):
         self.subtensor.substrate.submit_extrinsic = MagicMock(return_value = self.fail) 
-        assert False == bittensor.prometheus(wallet = self.wallet, subtensor = self.subtensor)
+        with pytest.raises(Exception):
+            bittensor.prometheus(wallet = self.wallet, subtensor = self.subtensor)