[hotfix] Autoscaler monitor fix unit tests

This commit is contained in:
Alex Wu
2020-08-27 14:26:41 -07:00
committed by GitHub
parent 76898d4ebc
commit 7dbc1f439c
+9 -10
View File
@@ -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