Skip to content

Commit f13f5f3

Browse files
authored
Merge pull request #2326 from opentensor/feat/roman/substrate-definition-before-initialization
add ConnectionRefusedError raising
2 parents 27bf3db + e0672f6 commit f13f5f3

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

bittensor/core/subtensor.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,7 @@ def __init__(
186186

187187
self.log_verbose = log_verbose
188188
self._connection_timeout = connection_timeout
189+
self.substrate: "SubstrateInterface" = None
189190
self._get_substrate()
190191

191192
def __str__(self) -> str:
@@ -201,7 +202,8 @@ def __repr__(self) -> str:
201202

202203
def close(self):
203204
"""Cleans up resources for this subtensor instance like active websocket connection and active extensions."""
204-
self.substrate.close()
205+
if self.substrate:
206+
self.substrate.close()
205207

206208
def _get_substrate(self):
207209
"""Establishes a connection to the Substrate node using configured parameters."""
@@ -223,14 +225,15 @@ def _get_substrate(self):
223225
except (AttributeError, TypeError, socket.error, OSError) as e:
224226
logging.warning(f"Error setting timeout: {e}")
225227

226-
except ConnectionRefusedError:
228+
except ConnectionRefusedError as error:
227229
logging.error(
228230
f"Could not connect to {self.network} network with {self.chain_endpoint} chain endpoint.",
229231
)
230232
logging.info(
231233
"You can check if you have connectivity by running this command: nc -vz localhost "
232-
f"{self.chain_endpoint.split(':')[2]}"
234+
f"{self.chain_endpoint}"
233235
)
236+
raise ConnectionRefusedError(error.args)
234237

235238
@staticmethod
236239
def config() -> "Config":

0 commit comments

Comments
 (0)