From bf030526551483f2afc66b941424913c88cdf698 Mon Sep 17 00:00:00 2001 From: Yadu Babuji Date: Mon, 30 Oct 2023 14:50:06 -0700 Subject: [PATCH] GlobusComputeEngine should use HTEX.scale_in which has better scale_in logic --- .../globus_compute_endpoint/engines/globus_compute.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/compute_endpoint/globus_compute_endpoint/engines/globus_compute.py b/compute_endpoint/globus_compute_endpoint/engines/globus_compute.py index 79910482e..73aa0ad46 100644 --- 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: """