GlobusComputeEngine should use HTEX.scale_in which has better scale_i…

…n logic
globus · Oct 30, 2023 · bf03052 · bf03052
1 parent 60ba518
commit bf03052
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/compute_endpoint/globus_compute_endpoint/engines/globus_compute.py b/compute_endpoint/globus_compute_endpoint/engines/globus_compute.py
@@ -133,8 +133,7 @@ def scale_out(self, blocks: int):
 
     def scale_in(self, blocks: int):
         logger.info(f"Scaling in {blocks} blocks")
-        to_kill = list(self.executor.blocks.values())[:blocks]
-        return self.provider.cancel(to_kill)
+        return self.executor.scale_in(blocks=blocks)
 
     def get_status_report(self) -> EPStatusReport:
         """