Skip to content

Commit b53e008

Browse files
committed
fix logic for handle node
1 parent 2da2e9c commit b53e008

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

scripts/launch-nodes/manage_subspace.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -150,15 +150,14 @@ def grep_protocol_version(client, retries=5, interval=30):
150150
def handle_node(client, node, subspace_dir, release_version, pot_external_entropy=None, plot_size=None, cache_percentage=None, network=None, prune=False, restart=False):
151151
"""Generic function to handle different node types with specified actions."""
152152
try:
153-
docker_compose_down(client, subspace_dir)
154-
modify_env_file(client, subspace_dir, release_version, pot_external_entropy=pot_external_entropy, plot_size=plot_size, cache_percentage=cache_percentage, network=network)
155-
docker_compose_up(client, subspace_dir)
156-
157153
if restart:
158154
docker_compose_restart(client, subspace_dir)
159-
160-
if prune:
155+
elif prune:
161156
docker_cleanup(client, subspace_dir)
157+
else:
158+
docker_compose_down(client, subspace_dir)
159+
modify_env_file(client, subspace_dir, release_version, pot_external_entropy=pot_external_entropy, plot_size=plot_size, cache_percentage=cache_percentage, network=network)
160+
docker_compose_up(client, subspace_dir)
162161

163162
except Exception as e:
164163
logger.error(f"Error handling node {node['host']}: {e}")

0 commit comments

Comments
 (0)