Refactor

MagicCastle · Jan 10, 2025 · 1fd8b66 · 1fd8b66
1 parent 6e43e44
commit 1fd8b66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/slurm_autoscale_tfe/__init__.py b/src/slurm_autoscale_tfe/__init__.py
@@ -169,7 +169,6 @@ def main(command, set_op, hostlist):
     hosts = frozenset(expand_hostlist(hostlist))
     tfe_client = connect_tfe_client()
     var_id, tfe_pool = get_pool_from_tfe(tfe_client)
-    tfe_instances, address_prefix = get_instances_from_tfe(tfe_client)
 
     next_pool = set_op(tfe_pool, hosts)
     if tfe_pool != next_pool:
@@ -186,6 +185,7 @@ def main(command, set_op, hostlist):
             hostlist,
         )
 
+    tfe_instances, address_prefix = get_instances_from_tfe(tfe_client)
     next_instances = set_op(tfe_instances, hosts)
     if tfe_instances != next_instances:
         try: