From 7dbc1f439c4e86ec50632eb59bc19390da65550c Mon Sep 17 00:00:00 2001 From: Alex Wu Date: Thu, 27 Aug 2020 14:26:41 -0700 Subject: [PATCH] [hotfix] Autoscaler monitor fix unit tests --- python/ray/monitor.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/python/ray/monitor.py b/python/ray/monitor.py index 7849cc1f7..72a962dac 100644 --- a/python/ray/monitor.py +++ b/python/ray/monitor.py @@ -100,16 +100,15 @@ class Monitor: """ waiting_bundles, infeasible_bundles = [], [] try: - if not self.autoscaler: - return - for resource_demand_pb in list( - resource_load_by_shape.resource_demands): - request_shape = dict(resource_demand_pb.shape) - for _ in range(resource_demand_pb.num_ready_requests_queued): - waiting_bundles.append(request_shape) - for _ in range( - resource_demand_pb.num_infeasible_requests_queued): - infeasible_bundles.append(request_shape) + if self.autoscaler: + for resource_demand_pb in list( + resource_load_by_shape.resource_demands): + request_shape = dict(resource_demand_pb.shape) + for _ in range(resource_demand_pb.num_ready_requests_queued): + waiting_bundles.append(request_shape) + for _ in range( + resource_demand_pb.num_infeasible_requests_queued): + infeasible_bundles.append(request_shape) except Exception as e: logger.exception(e) return waiting_bundles, infeasible_bundles