X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/3fa4a2b6138e3e9e468dd885a743ca38f08f0755..de4df7f80c531ab16e59ea36671a8efa9e6ff33d:/services/nodemanager/arvnodeman/jobqueue.py diff --git a/services/nodemanager/arvnodeman/jobqueue.py b/services/nodemanager/arvnodeman/jobqueue.py index 5ea2c5c8ee..f6e9249ebb 100644 --- a/services/nodemanager/arvnodeman/jobqueue.py +++ b/services/nodemanager/arvnodeman/jobqueue.py @@ -19,13 +19,14 @@ class ServerCalculator(object): """ class CloudSizeWrapper(object): - def __init__(self, real_size, **kwargs): + def __init__(self, real_size, node_mem_scaling, **kwargs): self.real = real_size for name in ['id', 'name', 'ram', 'disk', 'bandwidth', 'price', 'extra']: setattr(self, name, getattr(self.real, name)) self.cores = kwargs.pop('cores') self.scratch = self.disk + self.ram = int(self.ram * node_mem_scaling) for name, override in kwargs.iteritems(): if not hasattr(self, name): raise ValueError("unrecognized size field '%s'" % (name,)) @@ -42,8 +43,9 @@ class ServerCalculator(object): return True - def __init__(self, server_list, max_nodes=None, max_price=None): - self.cloud_sizes = [self.CloudSizeWrapper(s, **kws) + def __init__(self, server_list, max_nodes=None, max_price=None, + node_mem_scaling=0.95): + self.cloud_sizes = [self.CloudSizeWrapper(s, node_mem_scaling, **kws) for s, kws in server_list] self.cloud_sizes.sort(key=lambda s: s.price) self.max_nodes = max_nodes or float('inf') @@ -115,7 +117,7 @@ class JobQueueMonitorActor(clientactor.RemotePollLoopActor): queuelist = [] for out in squeue_out.splitlines(): cpu, ram, disk, reason, jobname = out.split(" ", 4) - if reason == "Resources": + if reason in ("Resources", "ReqNodeNotAvail"): queuelist.append({ "uuid": jobname, "runtime_constraints": {